Title: PRESTOPPING: HOW DOES EARLY STOPPING HELP GENERALIZATION AGAINST LABEL NOISE?

Abstract: Noisy labels are very common in real-world training data, which lead to poor generalization on test data because of overfitting to the noisy labels. In this paper, we claim that such overfitting can be avoided by "early stopping" training a deep neural network before the noisy labels are severely memorized. Then, we resume training the early stopped network using a "maximal safe set," which maintains a collection of almost certainly true-labeled samples at each epoch since the early stop point. Putting them all together, our novel two-phase training method, called Prestopping, realizes noise-free training under any type of label noise for practical use. Extensive experiments using four image benchmark data sets verify that our method significantly outperforms four state-of-the-art methods in test error by 0.4-8.2 percent points under existence of real-world noise.

1. INTRODUCTION:

By virtue of massive labeled data, deep neural networks (DNNs) have achieved a remarkable success in numerous machine learning tasks, such as image classification (Krizhevsky et al., 2012) and object detection (Redmon et al., 2016) . However, owing to their high capacity to memorize any label noise, the generalization performance of DNNs drastically falls down when noisy labels are contained in the training data (Jiang et al., 2018; Han et al., 2018; Song et al., 2019) . In particular, Zhang et al. (2017) have shown that a standard convolutional neural network (CNN) can easily fit the entire training data with any ratio of noisy labels and eventually leads to very poor generalization on the test data. Thus, it is challenging to train a DNN robustly even when noisy labels exist in the training data.

A popular approach to dealing with noisy labels is "sample selection" that selects true-labeled samples from the noisy training data (Jiang et al., 2018; Ren et al., 2018; Han et al., 2018; Yu et al., 2019; Song et al., 2019) . Here, (1-τ )×100% of small-loss training samples are treated as true-labeled ones and then used to update a DNN robustly, where τ ∈ [0, 1] is a noise rate. This loss-based separation is well known to be justified by the memorization effect (Arpit et al., 2017) that DNNs tend to learn easy patterns first and then gradually memorize all samples. In practice, Han et al. (2018) empirically proved that training on such small-loss samples yields a much better generalization performance under artificial noise scenarios.

Despite its great success, Song et al. (2019) have recently argued that the performance of the lossbased separation becomes considerably worse depending on the type of label noise. For instance, show those on CIFAR-100 with two types of synthetic noises of 40%, where "symmetric noise" flips a true label into other labels with equal probability, and "pair noise" flips a true label into a specific false label; (c) shows those on FOOD-101N (Lee et al., 2018) with real-world noise of 18.4%. in symmetric noise (Figure 1 (a)), the loss-based approach well separates true-labeled samples from false-labeled ones because many true-labeled ones exhibit smaller loss than false-labeled ones. On the other hand, in pair and real-world noises (Figures 1(b ) and 1(c)), many false-labeled samples are misclassified as true-labeled ones because the two distributions overlap closely; this overlap confirms that the loss-based separation still accumulates severe label noise from many misclassified cases, especially in real-world noise or pair noise which is regarded as more realistic than symmetric noise (Ren et al., 2018; Yu et al., 2019) . This limitation definitely calls for a new approach that supports any type of label noise for practical use.

In this regard, as shown in Figure 2 (a), we thoroughly investigated the memorization effect of a DNN on the two types of noises and found two interesting properties as follows:

• A noise type affects the memorization rate for false-labeled samples: The memorization rate for false-labeled samples is faster with pair noise than with symmetric noise. That is, the red portion in Figure 2 (a) starts to appear earlier in pair noise than in symmetric noise. This observation supports the significant overlap of true-labeled and false-labeled samples in Figure 1 (b). Thus, the loss-based separation performs well only if the false-labeled samples are scarcely learned at an early stage of training, as in symmetric noise.

• There is a period where the network accumulates the label noise severely: Regardless of the noise type, the memorization of false-labeled samples significantly increases at a late stage of training. That is, the red portion in Figure 2 (a) increases rapidly after the dashed line, in which we call the error-prone period. (See Section 3.2.1 for the details of estimating the error-prone period). We note that the training in that period brings no benefit. The generalization performance of "Default" deteriorates sharply, as shown in Figure 2 (c).

Based on these findings, we contend that eliminating this error-prone period should make a profound impact on robust optimization.

In this paper, we propose a novel approach, called Prestopping, that achieves noise-free training based on the early stopping mechanism. Because there is no benefit from the error-prone period, Prestopping early stops training before that period begins. This early stopping effectively prevents a network from overfitting to false-labeled samples, and the samples memorized until that point are added to a maximal safe set because they are true-labeled (i.e., blue in Figure 2 (a)) with high precision. Then, Prestopping resumes training the early stopped network only using the maximal safe set in support of noise-free training. Notably, our proposed merger of "early stopping" and "learning from the maximal safe set" indeed eliminates the error-prone period from the training process, as shown in Figure 2 (b). As a result, the generalization performance of a DNN remarkably improves in both noise types, as shown in Figure 2 (c).

To validate the superiority of Prestopping, DenseNet (Huang et al., 2017) and VGG-19 (Simonyan & Zisserman, 2015) were trained on both simulated and real-world noisy data sets, including CIFAR-10, CIFAR-100, ANIMAL-10N, and Food-101N. Compared with four state-of-the-art methods, Prestopping significantly improved test error by up to 18.1pp2 in a wide range of noise rates.

2. RELATED WORK:

Numerous studies have been conducted to address the problem of learning from noisy labels. A typical method is using "loss correction" that estimates the label transition matrix and corrects the loss of the samples in the mini-batch. Bootstrap (Reed et al., 2015) updates the network based on their own reconstruction-based objective with the notion of perceptual consistency. F-correction (Patrini et al., 2017) reweights the forward or backward loss of the training samples based on the label transition matrix estimated by a pre-trained normal network. R3: D2L (Ma et al., 2018) employs a simple measure called local intrinsic dimensionality and then uses it to modify the forward loss in order to reduce the effects of false-labeled samples in learning. Active Bias (Chang et al., 2017) heuristically evaluates uncertain samples with high prediction variances and then gives higher weights to their backward losses. Ren et al. (2018) include small clean validation data into the training data and re-weight the backward loss of the mini-batch samples such that the updated gradient minimizes the loss of those validation data. However, this family of methods is known to accumulate severe noise from the false correction, especially when the number of classes or the number of false-labeled samples is large (Han et al., 2018; Yu et al., 2019) .

To be free from the false correction, many recent researches have adopted "sample selection" that trains the network on selected samples. These methods attempt to select true-labeled samples from the noisy training data for use in updating the network. Decouple (Malach & Shalev-Shwartz, 2017) maintains two networks simultaneously and updates the models only using the samples that have different label predictions from these two networks. R3: Wang et al. (2018) proposed an iterative learning framework that learns deep discriminative features from well-classified noisy samples based on the local outlier factor algorithm (Breunig et al., 2000) . MentorNet (Jiang et al., 2018) introduces a collaborative learning paradigm that a pre-trained mentor network guides the training of a student network. Based on the small-loss criteria, the mentor network provides the student network with the samples whose label is probably correct. Co-teaching (Han et al., 2018) and Co-teaching+ (Yu et al., 2019 ) also maintain two networks, but each network selects a certain number of small-loss samples and feeds them to its peer network for further training. Compared with Co-teaching, Co-teaching+ further employs the disagreement strategy of Decouple. R1: ITLM (Shen & Sanghavi, 2019) iteratively minimizes the trimmed loss by alternating between selecting a fraction of small-loss samples at current moment and retraining the network using them. INCV (Chen et al., 2019) randomly divides the noisy training data and then utilizes cross-validation to classify true-labeled samples with removing large-loss samples at each iteration. However, their general philosophy of selecting small-loss samples works well only in some cases such as symmetric noise. RA: Differently to this family, we exploit the maximal safe set initially derived from the memorized samples at the early stop point.

Most recently, a hybrid of "loss correction" and "sample selection" approaches was proposed by Song et al. (2019) . Their algorithm called SELFIE trains the network on selectively refurbished false-labeled samples together with small-loss samples. SELFIE not only minimizes the number of falsely corrected samples but also exploits full exploration of the entire training data. Its component for sample selection can be replaced by our method to further improve the performance.

RA: For the completeness of the survey, we mention the work that provides an empirical or a theoretical analysis on why early stopping helps learn with the label noise. Oymak et al. (2019) and Hendrycks et al. (2019) have argued that early stopping is a suitable strategy because the network eventually begins to memorize all noisy samples if it is trained too long. Li et al. (2019) have theoretically proved that the network memorizes false-labeled samples at a later stage of training, and thus claimed that the early stopped network is fairly robust to the label noise. However, they did not mention how to take advantage of the early stopped network for further training. Please note that Prestopping adopts early stopping to derive a seed for the maximal safe set, which is exploited to achieve noise-free training during the remaining learning period. Thus, our novelty lies in the "merger" of early stopping and learning from the maximal safe set. , where ỹi ∈ {1, 2, . . . , k} is a noisy label which may not be true. Then, in conventional training, when a mini-batch B t = {x i , ỹi } b i=1 consists of b samples randomly drawn from the noisy training data D at time t, the network parameter θ t is updated in the descent direction of the expected loss on the mini-batch B t as in Eq. ( 1), where α is a learning rate and L is a loss function.

EQUATION

As for the notion of network memorization, a sample x is defined to be memorized by a network if the majority of its recent predictions at time t coincide with the given label, as in Definition 3.1.

Definition 3.1. (Memorized Sample) Let ŷt = Φ(x|θ t ) be the predicted label of a sample x at time t and H t x (q) = {ŷ t1 , ŷt2 , . . . , ŷtq } be the history of the sample x that stores the predicted labels of the recent q times, where Φ is a neural network. Next, P (y|x, t; q) is formulated such that it provides the probability of the label y ∈ {1, 2, ..., k} estimated as the label of the sample x based on H t x as in Eq. ( 2), where [•] is the Iverson bracket3 .

P (y|x, t; q) = ŷ∈H t x (q) [ŷ = y] |H t x (q)| (2)

Then, the sample x with its noisy label ỹ is a memorized sample of the network with the parameter θ t at time t if the condition in Eq. ( 3) holds.

EQUATION

3.2. TWO MAIN COMPONENTS AND PHASES:

The key idea of Prestopping is learning from a maximal safe set with an early stopped network. Thus, the two components of "early stopping" and "learning from the maximal safe set" respectively raise the questions about (Q1) when is the best point to early stop the training process? and (Q2) what is the maximal safe set to enable noise-free training during the remaining period?

3.2.1. QUESTION 1: BEST POINT TO EARLY STOP:

It is desirable to stop the training process at the point when the network (i) not only accumulates little noise from the false-labeled samples, (ii) but also acquires sufficient information from the true-labeled ones. Intuitively speaking, as indicated by the dashed line in Figure 3 (a), the best stop point is the moment when label precision and label recall in Definition 3.2 cross with each other because it is the best trade-off between the two metrics. The period beyond this point is what we call the error-prone period because label precision starts decreasing rapidly.

If the ground truth y * of a noisy label ỹ is known, the best stop point can be easily calculated by these metrics.

Definition 3.2. (Label Metrics) Let M t ⊆ D be a set of memorized samples at time t. Then, label precision (LP) and recall (LR) (Han et al., 2018) are formulated as Eq. ( 4).

LP = |{(x, ỹ) ∈ M t : ỹ = y * }| |M t | , LR = |{(x, ỹ) ∈ M t : ỹ = y * }| |{(x, ỹ) ∈ D : ỹ = y * }| (4)

However, it is not straightforward to find the exact best stop point without the ground-truth labels. Hence, we present two practical heuristics of approximating the best stop point with minimal supervision. These two heuristics require either a small clean validation set or a noise rate τ for the minimal supervision, where they are widely regarded as available in many studies (Veit et al., 2017; Ren et al., 2018; Han et al., 2018; Yu et al., 2019; Song et al., 2019) . • Validation Heuristic: If a clean validation set is given, we stop training the network when the validation error is the lowest. It is reasonable to expect that the lowest validation error is achieved near the cross point; after that point (i.e., in the error-prone period), the validation error likely increases because the network will be overfitted to the false-labeled samples. As shown in Figure 3 (b), the estimated stop point is fairly close to the best stop point in both noise types.

• Noise-Rate Heuristic: If a noise rate τ is known, we stop training the network when the training error reaches τ×100%. If we assume that all true-labeled samples of (1-τ )×100% are memorized before any false-labeled samples of τ ×100% (Arpit et al., 2017) , this point indicates the cross point of label precision and label recall with their values to be all 1. This heuristic tends to perform worse than the validation heuristic because the assumption does not hold perfectly. We further discuss its quality in Appendix A.

3.2.2. QUESTION 2: CRITERION OF A MAXIMAL SAFE SET:

Because the network is early stopped at the (estimated) best point, the set of memorized samples at that time is quantitatively sufficient and qualitatively less noisy. That is, it can be used as a safe and effective training set to resume the training of the early stopped network without accumulating the label noise. Based on this intuition, we define a maximal safe set in Definition 3.3, which is initially derived from the memorized samples at the early stop point and gradually increased as well as purified along with the network's learning progress. In each mini-batch B t , the network parameter θ t is updated using the current maximal safe set S t as in Eq. ( 5), and subsequently a more refined maximal safe set S t+1 is derived by the updated network.

Definition 3.3. (Maximal Safe Set) Let t stop be the early stop point. A maximal safe set S t at time t is defined to be the set of the memorized samples of the network Φ(x; θ t ) when t ≥ t stop . The network Φ(x; θ t ) at t = t stop is the early stopped network, i.e., S tstop = M tstop , and the network Φ(x; θ t ) at t > t stop is obtained by Eq. ( 5).

θ t+1 = θ t -α∇ 1 |B t | x∈B t L(x, ỹ; θ t ) B t = {x|x ∈ S t ∩ B t } (5)

Figure 4 shows the main advantage of learning from the maximal safe set during the remaining period. Even when the noise rate is quite high (e.g., 40%), this learning paradigm exploits most of true-labeled samples, in considering that the label recall of the maximal safe set was maintained very high in both CIFAR data sets regardless of the noise type. The label recall was maintained over 0.81 in symmetric noise and over 0.84 in pair noise even in CIFAR-100 with pair noise after the 80th epoch. Further, by excluding unsafe samples that might accumulate the label noise, the label precision of the maximal safe set tended to increase rapidly at the beginning of the remaining period; it was maintained over 0.98 in symmetric noise and over 0.94 even in CIFAR-100 with pair noise after the 80th epoch, which could not be realized by the small-loss trick (Han et al., 2018; Song et al., 2019; Shen & Sanghavi, 2019; Chen et al., 2019) . R2: In Section 5.3, we have empirically verified that the maximal safe set is superior to the selected samples based on the small-loss trick. Algorithm 1 describes the overall procedure of Prestopping with the validation heuristic, which is self-explanatory. First, the network is trained on the noisy training data D in the default manner (Lines 3-6). During this first phase, the validation data V is used to evaluate the best point for the early stop, and the network parameter is saved at the time of the lowest validation error (Lines 7-8). Then, during the second phase, Prestopping continues to train the early stopped network for the remaining learning period (Lines 10-12). Here, the maximal safe set S t at the current moment is retrieved, and each sample x ∈ S t ∩ B t is used to update the network parameter. The mini-batch samples not included in S t are no longer used in pursuit of robust learning (Lines 14-15).

4.2. COLLABORATION WITH SAMPLE REFURBISHMENT: Prestopping+:

To further improve the performance of Prestopping, we introduce Prestopping+ that employs the concept of selectively refurbishing false-labeled samples in SELFIE (Song et al., 2019) . In detail, the final maximal safe set S t end is retrieved from the first run of Prestopping, and then it is used as the true-labeled set for the next run of SELFIE with initializing the network parameter. Following the update principle of SELFIE, the modified gradient update rule in Eq. ( 6) is used to train the network. For each sample x in the mini-batch B t , the given label ỹ of x is selectively refurbished into y ref urb if it can be corrected with high precision. However, the mini-batch samples included in S t end are omitted from the refurbishment because their label is already highly credible. Then, the mini-batch samples in the refurbished set R t are provided to update the network together with those in the final maximal safe set S t end . Refer to Song et al. (2019) 's work for more details.

EQUATION

5. EVALUATION:

Data Sets: To verify the superiority of Prestopping, we performed an image classification task on four benchmark data sets: CIFAR-10 (10 classes)4 and CIFAR-100 (100 classes) 4 , a subset of 80 million categorical images (Krizhevsky et al., 2014) ; ANIMAL-10N (10 classes)5 , a real-world noisy data of human-labeled online images for confusing animals (Song et al., 2019) ; FOOD-101N (101 classes)6 , a real-world noisy data of crawled food images annotated by their search keywords in the FOOD-101 taxonomy (Bossard et al., 2014; Lee et al., 2018) . We did not apply any data augmentation.

Noise Injection: As all labels in the CIFAR data sets are clean, we artificially corrupted the labels in these data sets using typical methods for the evaluation of synthetic noises (Ren et al., 2018; Han et al., 2018; Yu et al., 2019) . For k classes, we applied the label transition matrix T: (i) symmetric noise:

∀ j =i T ij = τ k-1 and (ii) pair noise: ∃ j =i T ij = τ ∧ ∀ k =i,k =j T ik = 0

, where T ij is the probability of the true label i being flipped to the corrupted label j and τ is the noise rate. For the pair noise, the corrupted label j was set to be the next label of the true label i following the recent work (Yu et al., 2019; Song et al., 2019) . To evaluate the robustness on varying noise rates from light noise to heavy noise, we tested five noise rates τ ∈ {0.0, 0.1, 0.2, 0.3, 0.4}. In contrast, we did not inject any label noise into ANIMAL-10N and FOOD-101N7 because they contain real label noise estimated at 8.0% and 18.4% respectively (Lee et al., 2018; Song et al., 2019) .

Clean Validation Data: Recall that a clean validation set is needed for the validation heuristic in Section 3.2.1. As for ANIMAL-10N and Food-101N, we did exploit their own clean validation data; 5, 000 and 3, 824 images, respectively, were included in their validation set. However, as no validation data exists for the CIFAR data sets, we constructed a small clean validation set by randomly selecting 1, 000 images from the original training data of 50, 000 images. Please note that the noise injection process was applied to only the rest 49, 000 training images.

Networks and Hyperparameters: For the classification task, we trained DenseNet (L=40, k=12) and VGG-19 with a momentum optimizer. Specifically, we used a momentum of 0.9, a batch size of 128, a dropout of 0.1, and batch normalization. Prestopping has only one unique hyperparameter, the history length q, and it was set to be 10, which was the best value found by the grid search (see Appendix B.1 for details). The hyperparameters used in the compared algorithms were favorably set to be the best values presented in the original papers. As for the training schedule, we trained the network for 120 epochs and used an initial learning rate 0.1, which was divided by 5 at 50% and 75% of the total number of epochs.

Algorithms: We compared Prestopping and Prestopping+ with not only a baseline algorithm but also the four state-of-the-art robust training algorithms: Default trains the network without any processing for the label noise; Active Bias re-weights the backward loss of training samples based on their prediction variance; Co-teaching selects a certain number of small-loss samples to train the network based on the co-training (Blum & Mitchell, 1998) ; Co-teaching+ is similar to Co-teaching, but its small-loss samples are selected from the disagreement set; SELFIE selectively refurbishes noisy samples and exploits them together with the small-loss samples. All the algorithms were implemented Under review as a conference paper at ICLR 2020 Figure 6 : Best test errors using two CNNs on two data sets with varying symmetric noise rates.

using TensorFlow 1.8.0 and executed using 16 NVIDIA Titan Volta GPUs. For reproducibility, we provide the source code at https://bit.ly/2l3g9Jx. R2: In support of reliable evaluation, we repeated every task thrice and reported the average and standard error of the best test error, which was evaluated using the network at the time of the lowest validation error. Thus, the clean validation set was used to select the best model in all algorithms during the training procedure.

5.1. PERFORMANCE COMPARISON:

Figures 5 and 6 show the test error of the seven training methods using two CNNs on two data sets with varying pair and symmetric noise rates, respectively. The results of Prestopping and Prestopping+ in Section 5 were obtained using the validation heuristic. See Appendix A for the results of the noise-rate heuristic. In order to highlight the improvement of Prestopping and Prestopping+ over the other methods, their lines are dark colored. Figure 7 shows the test error on two real-world noisy data sets with different noise rates. The best test errors are detailed in Tables 1 and 2 by 0.3pp-17.5pp compared with the other robust methods. Unlike the pair noise, Co-teaching and SELFIE achieved a low test error comparable to Prestopping because true-labeled samples could be well separated from false-labeled ones by their small-loss criteria; hence, the dominance between Co-teaching and Active Bias was reversed so that the former slightly outperformed the latter.

5.1.3. RESULT WITH REAL-WORLD NOISE (FIGURE 7):

Both Prestopping and Prestopping+ maintained their dominance over the other methods under real-world label noise as well. Prestopping+ achieved the lowest test error when the number of classes is small (e.g., ANIMAL-10N), while Prestopping was the best when the number of classes is large (e.g., FOOD-101N) owing to the difficulty in label correction of Prestopping+. (Thus, practitioners are recommended to choose between Prestopping and Prestopping+ depending on the number of classes in hand.) Specifically, they improved the absolute test error by 0.4pp-4.6pp and 0.5pp-8.2pp in ANIMAL-10N and FOOD-101N, respectively. Therefore, we believe that the advantage of our methods is unquestionable even in the real-world scenarios.

5.2. CONVERGENCE ANALYSIS ON CLEAN VALIDATION DATA (FIGURE 8):

To verify that the error-prone period is eliminated, we plot the convergence of validation error for the seven training methods. The validation error of Default, Active Bias, and Co-teaching+ first reached their lowest values before the error-prone period, and then increased rapidly in the error-prone period. On the other hand, the remaining methods kept reducing the validation error even in that period. Notably, Prestopping+ achieved the lowest validation error at the end, owing to the successful merger of the advanced sample selection in Prestopping and the sample refurbishment in SELFIE. In addition, the performance of Prestopping was shown to be almost the same as and comparable to that of Prestopping+ in pair and symmetric noises, respectively. The validation error of SELFIE also converged faster than those of the other existing methods, but it was still inferior to Prestopping+. Thus, these results confirm that our two methods effectively overcome the error-prone period.

5.3. ACCURACY OF SAMPLE SELECTION (FIGURE 9):

RA: Figure 9 shows the average accuracy of sample selection for two different selection strategies using DenseNet. Average accuracy is defined as the average ratio of the true-labeled samples in selected mini-batch samples during the training process. As mentioned earlier in Section 1, because the loss distribution of false-labeled samples was overlapped closely with that of true-labeled ones in pair noise, the average accuracy of the small-loss trick was worse with pair noise than with symmetric noise; it decreased to 75.0% as the noise rate increased in CIFAR-100 with pair noise. On the other hand, the maximal safe set maintained its dominance over the small-loss trick in all data sets regardless of the noise type. In particular, the average accuracy of the maximal safe set was 93.2% 

5.4. ANATOMY OF Co-teaching+ (FIGURE 10):

Although Co-teaching+ is the latest method, its performance was worse than expected, as shown in Section 5.1. Thus, we looked into Co-teaching+ in more detail. A poor performance of Co-teaching+ was attributed to the fast consensus of the label predictions for true-labeled samples, especially when training a complex network. In other words, because two complex networks in Co-teaching+ start making the same predictions for true-labeled samples too early, these samples are excluded too early.

As shown in Figures 10(a ) and 10(b), the disagreement ratio with regard to the true-labeled samples dropped faster with a complex network than with a simple network, in considering that the ratio drastically decreased to 49.8% during the first 5 epochs. Accordingly, it is evident that Co-teaching+ with a complex network causes a narrow exploration of the true-labeled samples, and the selection accuracy of Co-teaching+ naturally degraded from 60.4% to 44.8% for that reason, as shown in Figure 10(c ). Therefore, we conclude that Co-teaching+ may not suit a complex network.

6. CONCLUSION:

In this paper, we proposed a novel two-phase training strategy for the noisy training data, which we call Prestopping. The first phase, "early stopping," retrieves an initial set of true-labeled samples as many as possible, and the second phase, "learning from a maximal safe set," completes the rest training process only using the true-labeled samples with high precision. Prestopping can be easily applied to many real-world cases because it additionally requires only either a small clean validation set or a noise rate. Furthermore, we combined this novel strategy with sample refurbishment to develop Prestopping+. (FIGURE 11) To verify the performance of Prestopping and Prestopping+ with the noise-rate heuristic in Section 3.2.1, we trained a VGG-19 network on two simulated noisy data sets with the same configuration as in Section 5. Figure 11 shows the test error of our two methods using the noise-rate heuristic as well as those of the other five training methods. Again, the test error of either Prestopping or Prestopping+ was the lowest at most error rates with any noise type. The trend of the noise-rate heuristic here was almost the same as that of the validation heuristic in Section 5.1. Especially when the noise rate was 40%, Prestopping and Prestopping+ significantly improved the test error by 5.1pp-17.0pp in the pair noise (Figure 11 A.2 COMPARISON WITH VALIDATION HEURISTIC (FIGURE 12 )

Figure 12 shows the difference in test error caused by the two heuristics. Overall, the performance with the noise-rate heuristic was worse than that with validation heuristic, even though the worse one outperformed the other training methods as shown in Figure 11 . As the assumption of the noise-rate heuristic does not hold perfectly, a lower performance of the noise-rate heuristic looks reasonable. However, we expect that the performance with this heuristic can be improved by stopping a little earlier than the estimated point, and we leave this extension as the future work. Prestopping requires one additional hyperparameter, the history length q in Definition 3.1. For hyperparameter tuning, we trained DenseNet (L=40, k=12) on CIFAR-10 and CIFAR-100 with a noise rate of 40%, and the history length q was chosen in a grid ∈ {1, 5, 10, 15, 20}. Figure 13 shows the test error of Prestopping obtained by the grid search on two noisy CIFAR data sets. Regardless of the noise type, the lowest test error was typically achieved when the value of q was 10 in both data sets. Therefore, the history length q was set to be 10 in all experiments. Figure 13 : Grid search on CIFAR-10 and CIFAR-100 with two types of noises of 40%.

B.2 EXPERIMENT USING TINY-IMAGENET DATA SET (FIGURE 14 )

R3: For a larger-scale experiment, we repeated the image classification task on Tiny-ImageNet (200 classes), a subset of ImageNet (Krizhevsky et al., 2012) , with 100, 000 training and 10, 000 validation images. Because no test set exists, randomly selected 9, 000 images from the validation set were used as the test set, and the rest 1, 000 validation images were used as the clean validation set for the validation heuristic. The experimental configurations were the same as those in Section 5.

Figure 14 shows the test error of the seven training methods using VGG-19 on Tiny-ImageNet with varying noise rates. Similar to CIFAR data sets in Figures 5 and 6 that there was no synergistic effect from collaboration with sample refurbishment because of the larger number of classes (i.e., 200). In particular, compared with other robust methods, Prestopping showed significant reduction in the absolute test error by 7.8pp-12.7pp at pair noise of 40% and 1.7pp-13.9pp at symmetric noise of 40%.

B.3 EXPERIMENT USING CLOTHING DATA SET (FIGURE 15 )

R2: We additionally evaluated the superiority of Prestopping(+) on another challenging real-world noisy data set Clothing70k (14 classes), a subset 8 of Clothing1M (Xiao et al., 2015) , where its noise rate was estimated at 38.5%. For the Clothing70k data set, we randomly selected 70, 000 images from 1 million noisy training images in Clothing1M and used them as its noisy training set; we did exploit the original clean validation and test sets consisting of 14, 313 and 10, 526 images, respectively. The experimental configurations were the same as those in Section 5.

Figure 15 B.4 IMPACT OF NUMBER OF CLASSES ON Prestopping+ (FIGURE 16 )

We investigate the impact of the number of classes on Prestopping+ under the pair noise of 40%.

Figure 16 shows the ratio of refurbished samples used for training (hatched bar) with the refurbishing accuracy (solid line) on two data sets with 10 and 100 classes. When the number of classes is small (e.g., 10), Prestopping+ refurbished most of false-labeled samples very accurately; the refurbishing accuracy was consistently over 91.6% during the entire training period. On the other hand, when the number of classes is large (e.g., 100), the refurbishing accuracy drastically dropped from 92.0% to 76.7% as more samples were refurbished. Thus, as shown in Figure 16 (b), collaboration with sample refurbishment is not always synergistic because such a significant drop in the refurbishing accuracy could induce many falsely corrected samples, especially when there are many classes. 

C COMPLETE RESULTS ON BEST TEST ERROR:



:

The learning rate, as usual, was decayed at 50% and 75% of the total number of training epochs.

A pp is the abbreviation of a percentage point.

The Iverson bracket [P ] returns 1 if P is true; 0 otherwise.

https://www.cs.toronto.edu/~kriz/cifar.html

https://dm.kaist.ac.kr/datasets/animal-10n

https://kuanghuei.github.io/Food-101N

In FOOD-101N, we used a subset of the entire training data marked with whether the label is correct or not.

8R2: We used a subset of Clothing1M to finish this experiment within the ICLR'20 rebuttal period.

Figures and Tables:

Symmetric Noise. (b) Pair Noise.(c) Real-World Noise.

Figure 1: Loss distributions at the training accuracy of 50% using DenseNet (L=40, k=12): (a) and (b) show those on CIFAR-100 with two types of synthetic noises of 40%, where "symmetric noise" flips a true label into other labels with equal probability, and "pair noise" flips a true label into a specific false label; (c) shows those on FOOD-101N (Lee et al., 2018) with real-world noise of 18.4%.

Figure 2: Key idea of Prestopping: (a) and (b) show how many true-labeled and false-labeled samples are memorized when training DenseNet (L=40, k=12) 1 on CIFAR-100 with two types of synthetic noises of 40%. "Default" is a standard training method, and "Prestopping" is our proposed one; (c) contrasts the convergence of test error between the two methods.

Figure 3: Early stop point estimated by ideal and heuristic methods when training DenseNet (L=40, k=12) on CIFAR-100 with two types of synthetic noises of 40%: (a) and (b) show the stop point derived by the ground-truth labels and the clean validation set, respectively.

R2: Label precision and label recall on CIFAR-10 (10 classes). Label precision and label recall on CIFAR-100 (100 classes).

Figure 4: Label precision and label recall of the maximal safe set during the remaining epochs when training DenseNet (L=40, k=12) on CIFAR data sets with two types of synthetic noises of 40%.

Figure 8: Convergence curves of DenseNet (L=40, k=12) on CIFAR-10 with a noise rate of 40%.

Figure 9: Average accuracy of selecting true-labeled samples by the small-loss trick in Co-teaching and the maximal safe set in Prestopping using DenseNet on two data sets with synthetic noises.

Figure 10: Anatomy of Co-teaching+ on CIFAR-100 with 40% symmetric noise: (a) and (b) showthe change in disagreement ratio for all true-labeled samples, when using Co-teaching+ to train two networks with different complexity, where "simple network" is a network with seven layers used byYu et al. (2019), and "complex network" is a DenseNet (L=40, k=12) used for our evaluation; (c) shows the accuracy of selecting true-labeled samples on the DenseNet.

(a)) and 0.3pp-17.3pp in the symmetric noise (Figure 11(b)) compared with the other robust methods.

Figure 11: Best test errors using VGG-19 on two simulated noisy data sets with varying noise rates.

a) Pair Noise. (b) Symmetric Noise.

Figure 12: Difference in test error between two heuristics on two simulated noisy data sets. B SUPPLEMENTARY EVALUATION B.1 HYPERPARAMETER SELECTION (FIGURE 13)

Figure 14: Best test errors using VGG-19 on Tiny-ImageNet with varying noise rates.

Figure 15: Best test errors on Clothing70k (τ ≈ 38.5%) along with the detailed result.

Figure 16: Refurbishing ratio and accuracy of Prestopping+ when training DenseNet (L=40, k=12) on CIFAR-10 and CIFAR-100 with pair noise of 40%.

4: Label precision and label recall of the maximal safe set during the remaining epochs when training DenseNet (L=40, k=12) on CIFAR data sets with two types of synthetic noises of 40%.

Table content: <table><tr><td>13:</td><td colspan="3">Draw a mini-batch B t from D;</td></tr><tr><td>14:</td><td colspan="3">S t ← {x|argmax y P (y|x, t; q) = ỹ}; /* A maximal safe set in Definition 3.3 */</td></tr><tr><td>15:</td><td>θ t+1 = θ t -α∇</td><td>1 |St∩Bt|</td><td>x∈St∩Bt L(x, ỹ; θ t ) ; /* Update by Eq. (5) */</td></tr><tr><td>16:</td><td>t ← t + 1;</td><td/><td/></tr><tr><td colspan="2">17: return θ t , S t ;</td><td/><td/></tr><tr><td colspan="4">4 ALGORITHM DESCRIPTION</td></tr><tr><td colspan="4">4.1 MAIN ALGORITHM: Prestopping WITH VALIDATION HEURISTIC</td></tr></table>

The performance trend in the two network architectures was similar with each other. In general, either Prestopping or Prestopping+ achieved the lowest test error in a wide range of noise rates on both CIFAR data sets. With help of the refurbished samples, Prestopping+ achieved a slightly better performance than Prestopping in CIFAR-10. However, an opposite trend was observed in CIFAR-100; this phenomenon was due to a large number of falsely corrected labels under the pair noise, especially when the number of classes is large (see Appendix B.4 for details). Although SELFIE achieved relatively lower test error among the existing methods, the test error of SELFIE was still worse than that of Prestopping. Co-teaching did not work well because many false-labeled samples were misclassified as clean ones; Co-teaching+ was shown to be even worse than Co-teaching despite it being an improvement of Co-teaching (see Section 5.4 for details). The test error of Active Bias was not comparable to that of Prestopping. The performance improvement of ours over the others increased as the label noise became heavier. In particular, at a heavy noise rate of 40%, Prestopping or Prestopping+ significantly reduced the absolute test error by 2.2pp-18.1pp compared with the other robust methods. Figure 7: Best test errors using two CNNs on two data sets with real-world noises.

Table content: <table><tr><td/><td colspan="2">Default</td><td colspan="2">Active Bias</td><td colspan="2">Co-teaching</td><td colspan="3">Co-teaching+</td><td>SELFIE</td><td>LESSON</td><td>LESSON+</td></tr><tr><td/><td>Default</td><td colspan="2">Active Bias</td><td colspan="2">Co-teaching</td><td colspan="2">Co-teaching+</td><td colspan="2">SELFIE</td><td colspan="2">Prestopping (Validation)</td><td>Prestopping+ (Validation)</td></tr><tr><td/><td>18.3%</td><td/><td/><td>21.1%</td><td/><td/><td/><td/><td>57.7%</td><td/><td>72.5%</td></tr><tr><td>Error</td><td>17.3%</td><td colspan="2">DenseNet</td><td>19.9%</td><td/><td/><td>VGG</td><td>Error</td><td>56.4%</td><td colspan="2">DenseNet</td><td>70.5%</td><td>VGG</td></tr><tr><td>Test</td><td/><td/><td/><td/><td/><td/><td/><td>Test</td><td/><td/></tr><tr><td>Best</td><td>16.3%</td><td/><td/><td>18.7%</td><td/><td/><td/><td>Best</td><td>55.1%</td><td/><td>68.5%</td></tr><tr><td/><td>15.3%</td><td/><td/><td>17.5%</td><td/><td/><td/><td/><td>53.8%</td><td/><td>66.5%</td></tr><tr><td/><td/><td>Method</td><td/><td/><td/><td>Method</td><td/><td/><td/><td>Method</td><td>Method</td></tr><tr><td/><td colspan="6">(a) ANIMAL-10N (τ ≈ 8.0%).</td><td/><td/><td/><td colspan="2">(b) FOOD-101N (τ ≈ 18.4%).</td></tr><tr><td/><td>47%</td><td/><td/><td colspan="3">Error-prone period</td><td>47%</td><td/><td/><td colspan="2">Error-prone period</td><td>Default</td></tr><tr><td>Validation Error</td><td>21% 34%</td><td/><td/><td/><td/><td/><td>21% 34%</td><td/><td/><td/><td>Active Bias Co-teaching Co-teaching+ SELFIE</td></tr><tr><td/><td>8%</td><td/><td/><td/><td/><td/><td>8%</td><td/><td/><td/><td>Prestopping</td></tr><tr><td/><td>0</td><td>30</td><td>60</td><td>90</td><td/><td>120</td><td>0</td><td/><td>30</td><td>60</td><td>90</td><td>120</td><td>Prestopping+</td></tr><tr><td/><td/><td/><td>Epoch</td><td/><td/><td/><td/><td/><td/><td>Epoch</td></tr></table>

Through extensive experiments using various real-world and simulated noisy data sets, we verified that either Prestopping or Prestopping+ achieved the lowest test error among the seven compared methods, thus significantly improving the robustness to diverse types of label noise. Overall, we believe that our work of dividing the training process into two phases by early stopping is a new direction for robust training and can trigger a lot of subsequent studies.A Prestopping WITH NOISE-RATE HEURISTIC Algorithm 2 describes the overall procedure of Prestopping with the noise-rate heuristic, which is also self-explanatory. Compared with Algorithm 1, only the way of determining the best stop point in Lines 7-9 has changed. 11: θ t ← θ tstop ; /* Load the network stopped at t stop */ 12: for i = stop_epoch to epochs do /* Phase II: Learning from a maximal safe set */ S t ← {x|argmax y P (y|x, t; q) = ỹ}; /* A maximal safe set in Definition 3.3 */

Table content: <table><tr><td colspan="4">Algorithm 2 Prestopping with Noise-Rate Heuristic</td></tr><tr><td colspan="4">INPUT: D: data, epochs: total number of epochs, q: history length, τ : noise rate</td></tr><tr><td colspan="3">OUTPUT: θ t : network parameters</td><td/></tr><tr><td colspan="4">1: t ← 1; θ t ← Initialize the network parameter;</td></tr><tr><td colspan="4">2: θ tstop ← ∅; /* The parameter of the stopped network */</td></tr><tr><td colspan="4">3: for i = 1 to epochs do /* Phase I: Learning from a noisy training data set */</td></tr><tr><td>4:</td><td colspan="2">for j = 1 to | D|/|B t | do</td><td/></tr><tr><td>5: 6:</td><td colspan="3">Draw a mini-batch B t from D; θ t+1 = θ t -α∇ 1 |Bt| x∈Bt L(x, ỹ; θ t ) ; /* Update by Eq. (1) */</td></tr><tr><td>7:</td><td colspan="3">train_err ← Get_Training_Error( D, θ t ); /* A training error at time t */</td></tr><tr><td>8:</td><td colspan="3">if train_err ≤ τ then /*Save the network when train_err ≤ τ */</td></tr><tr><td>9:</td><td colspan="2">θ tstop ← θ t ; break;</td><td/></tr><tr><td>10:</td><td>t ← t + 1;</td><td/><td/></tr><tr><td>13:</td><td colspan="2">for j = 1 to | D|/|B t | do</td><td/></tr><tr><td>14:</td><td colspan="3">Draw a mini-batch B t from D;</td></tr><tr><td>15:</td><td/><td/><td/></tr><tr><td>16:</td><td>θ t+1 = θ t -α∇</td><td>1 |St∩Bt|</td><td>x∈St∩Bt L(x, ỹ; θ t ) ; /* Update by Eq. (5) */</td></tr><tr><td>17:</td><td>t ← t + 1;</td><td/><td/></tr><tr><td colspan="2">18: return θ t , S t ;</td><td/><td/></tr><tr><td colspan="4">A.1 RESULT WITH SYNTHETIC NOISE</td></tr></table>

, both Prestopping and Prestopping+ outperformed the other robust methods in both noise types. The only difference was

Table content: <table><tr><td/><td/><td>Default</td><td colspan="2">Active Bias</td><td colspan="2">Co-teaching</td><td colspan="2">Co-teaching+</td><td>SELFIE</td><td>LESSON</td><td>LESSON+</td></tr><tr><td/><td/><td>Default</td><td>Active Bias</td><td colspan="2">Co-teaching</td><td colspan="2">Co-teaching+</td><td>SELFIE</td><td>Prestopping (Validation)</td><td>Prestopping+ (Validation)</td></tr><tr><td>Best Test Error</td><td>11% 16% 21% 60% 67% 53%</td><td>VGG</td><td/><td>53% 61% 69%</td><td>VGG</td><td/><td/><td>Noise Type Default Active Bias Co-teaching Co-teaching+</td><td>Pair Noise 65.6±0.52 62.9±0.31 67.6±0.22 67.8±0.31</td><td>Symmetric Noise 68.3±1.02 66.7±0.87 64.0±0.41 75.4±0.59</td></tr><tr><td/><td>6% 46%</td><td colspan="2">0% 10% 20% 30% 40% 0% 10% 20% 30% 40% Noise Rate</td><td>45%</td><td colspan="3">0% 10% 20% 30% 40% Noise Rate</td><td>SELFIE Prestopping Prestopping+</td><td>63.6±0.44 55.1±0.24 56.4±0.37</td><td>63.2±0.19 61.5±0.31 62.8±0.25</td></tr><tr><td/><td/><td colspan="2">(a) Pair Noise.</td><td/><td colspan="3">(b) Symmetric Noise.</td></tr></table>

shows the test error of the seven training methods using two CNNs on Clothing70k with real-world noise of 38.5%. Overall, Prestopping and Prestopping+ significantly outperformed the other robust methods. In particular, Prestopping+ achieved the lowest test error because the number of classes was not that large (i.e., 14 classes). Compared with other state-of-the-art methods, it improved the absolute test error by 1.1pp-3.3pp using DenseNet and 0.7pp-5.8pp using VGG-19.

Table content: <table><tr><td/><td/><td>Default</td><td colspan="2">Active Bias</td><td colspan="2">Co-teaching</td><td>Co-teaching+</td><td>SELFIE</td><td>LESSON</td><td>LESSON+</td></tr><tr><td/><td/><td>Default</td><td>Active Bias</td><td colspan="2">Co-teaching</td><td>Co-teaching+</td><td>SELFIE</td><td>Prestopping (Validation)</td><td>Prestopping+ (Validation)</td></tr><tr><td/><td>39.0%</td><td/><td/><td>47.2%</td><td/><td/><td/><td>Network</td><td>DenseNet</td><td>VGG-19</td></tr><tr><td>Error</td><td>38.0%</td><td/><td>DenseNet</td><td>45.3%</td><td/><td colspan="2">VGG</td><td>Default Active Bias</td><td>38.8±0.10 37.7±0.09</td><td>46.9±0.31 46.5±0.70</td></tr><tr><td>Test Best</td><td>37.0%</td><td/><td/><td>43.4%</td><td/><td/><td/><td>Co-teaching Co-teaching+ SELFIE</td><td>37.6±0.14 39.7±0.18 37.5±0.14</td><td>43.3±0.34 47.8±0.51 42.7±0.13</td></tr><tr><td/><td>36.0%</td><td colspan="2">Method</td><td>41.5%</td><td/><td>Method</td><td/><td>Prestopping Prestopping+</td><td>36.6±0.11 36.4±0.07</td><td>42.2±0.12 42.0±0.11</td></tr><tr><td/><td/><td colspan="2">(a) DenseNet.</td><td/><td colspan="2">(b) VGG-19.</td><td/><td>(c) Detailed Results.</td></tr></table>

shows the test error of seven training methods using two CNNs on two simulated noisy data sets with varying noise rates. Table2shows the test error of seven training methods using two CNNs on two real-world noisy data sets with different noise rates.

Table content: <table/>

The best test errors (%) of seven training methods on two types of synthetic noises with varying noise rates (0%, 10%, 20%, 30%, and 40%) in Figures5 and 6.

Table content: <table><tr><td>Noise Type</td><td/><td colspan="2">Pair Noise in Figure 5</td><td/><td/><td colspan="2">Symmetric Noise in Figure 6</td><td/></tr><tr><td>Data Set</td><td colspan="2">CIFAR-10</td><td colspan="2">CIFAR-100</td><td colspan="2">CIFAR-10</td><td colspan="2">CIFAR-100</td></tr><tr><td>Network</td><td>DenseNet</td><td>VGG</td><td>DenseNet</td><td>VGG</td><td>DenseNet</td><td>VGG</td><td>DenseNet</td><td>VGG</td></tr></table>

Peer review (iclr2020):
This paper proposes a two-phase training method for learning with label noise. On the positive side, this paper focuses on the idea of prestopping and proposes several relevant definitions to formalize their idea and come up with a heuristic algorithm. However, I believe the paper has missed several very relevant papers that provides very similar ideas. Both [2] & [3] provide theoretical analysis to why early stopping matters in learning with label noise for DNNs. Before these two papers, [1] also observed that the learning trajectories for clean and noisy samples are different in label noise problem, and they used early stopping in their experiments to address this issue. Given these existing literatures, the contribution of this paper should be considered more properly. [1] Learning with Bad Training Data via Iterative Trimmed Loss Minimization, Yanyao Shen, Sujay Sanghavi, ICML 2019. [2] Hu, Wei, Zhiyuan Li, and Dingli Yu. *Understanding Generalization of Deep Neural Networks Trained with Noisy Labels.* arXiv preprint arXiv:1905.11368 (2019). [3] Li, Mingchen, Mahdi Soltanolkotabi, and Samet Oymak. *Gradient descent with early stopping is provably robust to label noise for overparameterized neural networks.* arXiv preprint arXiv:1903.11680 (2019).