8일 전

소음 있는 레이블 극복을 위한 합의 기반 접근: 공정규화를 통한 공동 학습 방법

Hongxin Wei, Lei Feng, Xiangyu Chen, Bo An
소음 있는 레이블 극복을 위한 합의 기반 접근: 공정규화를 통한 공동 학습 방법
초록

노이즈가 있는 레이블을 가진 데이터를 활용한 딥러닝은 약한 지도 학습 분야에서 실질적으로 도전적인 문제이다. 최신 기법인 '디커플링(Decoupling)'과 '코티칭+ (Co-teaching+)'은 노이즈 레이블 학습 문제 완화에 있어 '불일치(Disagreement)' 전략이 핵심적이라고 주장한다. 본 논문에서는 이러한 관점과는 다른 접근을 취하여, 학습 도중 두 개의 네트워크 간의 다양성을 줄이는 강건한 학습 프레임워크인 JoCoR(Joining and Co-Regularization)를 제안한다. 구체적으로, 동일한 미니배치 데이터에 대해 두 개의 네트워크가 예측을 수행하고, 각 학습 예제에 대해 Co-Regularization을 활용한 공동 손실(joint loss)을 계산한다. 이후 작은 손실을 가진 예제들을 선택하여 두 네트워크의 파라미터를 동시에 업데이트한다. 공동 손실에 의해 학습된 두 네트워크는 Co-Regularization의 영향으로 점점 더 유사해지게 된다. MNIST, CIFAR-10, CIFAR-100, Clothing1M 등의 벤치마크 데이터셋에서 노이즈가 포함된 손상된 데이터를 활용한 광범위한 실험 결과를 통해, JoCoR가 노이즈 레이블 학습에 있어 많은 최신 기법들보다 우수함을 입증하였다.