US 12,205,349 B2
System and method for improving robustness of pretrained systems in deep neural networks utilizing randomization and sample rejection
Fatemeh Sheikholeslami, Pittsburgh, PA (US); Wan-Yi Lin, Wexford, PA (US); Jan Hendrik Metzen, Boeblingen (DE); Huan Zhang, Pittsburgh, PA (US); and Jeremy Kolter, Pittsburgh, PA (US)
Assigned to Robert Bosch GmbH, (DE); and Carnegie Mellon University, Pittsburgh, PA (US)
Filed by Robert Bosch GmbH, Stuttgart (DE); and Carnegie Mellon University, Pittsburgh, PA (US)
Filed on Mar. 18, 2022, as Appl. No. 17/698,556.
Prior Publication US 2023/0298315 A1, Sep. 21, 2023
Int. Cl. G06V 10/764 (2022.01); G06T 5/70 (2024.01); G06V 10/84 (2022.01)
CPC G06V 10/764 (2022.01) [G06T 5/70 (2024.01); G06V 10/84 (2022.01); G06T 2207/20076 (2013.01); G06T 2207/20081 (2013.01); G06T 2207/20084 (2013.01)] 19 Claims
OG exemplary drawing
 
1. A computer-implemented method for training a machine-learning network, comprising:
receiving an input data from a sensor, wherein the input data is indicative of image, radar, sonar, or sound information;
generating a perturbed input data set utilizing the input data, wherein the perturbed input data set includes perturbed data;
denoising, via a pre-trained denoiser, the perturbed input data set to generate a denoised data set;
training the machine-learning network utilizing the denoised data set, wherein the machine-learning network is configured to reject the denoised data set, utilizing a rejector, when a classification probability falls below a classification threshold, wherein the classification threshold is associated with classification the denoised data set, wherein the rejector is trained to discriminate between correctly classified inputs and misclassified denoised inputs; and
in response to the classification probability falling below the classification threshold, outputting an abstain classification associated with the input data, wherein the abstain classification is ignored for classifying.