노이즈 레이블 수정 및 견고한 장면 그래프 생성을 위한 훈련

Konsep Inti

기존 장면 그래프 생성 모델들은 데이터셋의 노이즈 레이블 문제를 간과해왔다. 이 논문에서는 노이즈 레이블 문제를 해결하기 위해 노이즈 레이블 수정 및 견고한 훈련 전략인 NICEST를 제안한다.

Abstrak

이 논문은 장면 그래프 생성(SGG) 문제에서 노이즈 레이블 문제를 다룬다. 기존 SGG 모델들은 다음과 같은 두 가지 가정을 전제로 하고 있다: 1) 모든 수동 주석된 긍정 샘플은 equally correct하다. 2) 모든 비주석된 부정 샘플은 절대적으로 배경이다.

그러나 저자들은 이 두 가정이 SGG에 적용되지 않는다고 주장한다. 실제로 SGG 데이터셋에는 "noisy" 관계 레이블이 많이 존재하며, 이는 모델 훈련을 편향시킨다.

이를 해결하기 위해 저자들은 NICEST(NoIsy label CorrEction and Sample Training)를 제안한다. NICEST는 두 부분으로 구성된다:

1. NICE(NoIsy label CorrEction): NICE는 노이즈 샘플을 탐지하고 더 높은 품질의 soft 관계 레이블을 할당한다. 구체적으로 NICE는 다음 3단계로 구성된다:

• Neg-NSD: 부정 노이즈 샘플 탐지. 이를 out-of-distribution 탐지 문제로 다룸.
• Pos-NSD: 긍정 노이즈 샘플 탐지. 밀도 기반 클러스터링 알고리즘 사용.
• NSC: 노이즈 샘플 레이블 보정. 가중치 KNN 사용하여 soft 레이블 할당.

2. NIST(NoIsy Sample Training): NIST는 다중 교사 지식 증류 기반 훈련 전략으로, 편향된 지식을 완화하여 모델이 편향되지 않은 지식을 학습할 수 있게 한다. 동적 가중치 전략을 사용하여 다양한 교사 모델의 편향을 제어한다.

또한 저자들은 기존 SGG 벤치마크의 한계를 극복하기 위해 VG-OOD라는 새로운 벤치마크를 제안했다. VG-OOD는 훈련 세트와 테스트 세트 간 관계 분포를 의도적으로 다르게 만들어, 모델의 일반화 능력을 더 잘 평가할 수 있다.

실험 결과, NICEST는 다양한 SGG 모델과 데이터셋에서 효과적이고 일반화 능력이 뛰어난 것으로 나타났다.

Statistik

기존 SGG 데이터셋의 관계 레이블 분포는 head 카테고리가 tail 카테고리보다 38배 더 많다.
VG-OOD 데이터셋의 훈련 세트와 테스트 세트 간 관계 분포 KL divergence는 7.55로, 원본 VG의 0.82에 비해 크게 증가했다.

Kutipan

"Nearly all existing SGG models have overlooked the ground-truth annotation qualities of mainstream SGG datasets, i.e., they assume: 1) all the manually annotated positive samples are equally correct; 2) all the un-annotated negative samples are absolutely background."
"We argue that neither of the assumptions applies to SGG: there are numerous "noisy" ground-truth predicate labels that break these two assumptions and harm the training of unbiased SGG models."

Wawasan Utama Disaring Dari

NICEST: Noisy Label Correction and Training for Robust Scene Graph Generation

by Lin Li,Jun X... pada arxiv.org 04-16-2024

https://arxiv.org/pdf/2207.13316.pdf

Pertanyaan yang Lebih Dalam

SGG 모델의 성능 향상을 위해 어떤 다른 접근법을 고려해볼 수 있을까

다른 SGG 모델의 성능 향상을 위해 고려할 수 있는 다른 접근법은 데이터 증강 기술을 활용하는 것입니다. 데이터 증강은 기존 데이터를 변형하거나 확장하여 모델의 일반화 성능을 향상시키는 데 도움이 될 수 있습니다. 예를 들어, 이미지 회전, 반전, 크기 조정, 색상 변환 등의 기술을 사용하여 데이터를 다양하게 만들어 모델이 다양한 시나리오에 대해 더 잘 대응할 수 있도록 할 수 있습니다. 또한, 생성적 적대 신경망(GAN)을 활용하여 실제와 유사한 합성 이미지를 생성하여 데이터셋을 보강하는 방법도 효과적일 수 있습니다.

기존 SGG 데이터셋의 노이즈 레이블 문제를 해결하는 다른 방법은 무엇이 있을까

기존 SGG 데이터셋의 노이즈 레이블 문제를 해결하는 다른 방법으로는 자동 주석 생성 기술을 활용하는 것이 있습니다. 자동 주석 생성은 이미지나 비디오에 대한 자동 주석을 생성하는 기술로, 레이블이 부족하거나 잘못된 경우에도 모델이 학습할 수 있는 더 많은 데이터를 생성할 수 있습니다. 또한, 준지도 학습 기술을 활용하여 레이블이 부족한 데이터에 대해 모델을 학습시키는 방법도 노이즈 레이블 문제를 완화하는 데 도움이 될 수 있습니다.

NICEST의 아이디어를 다른 비전 및 언어 이해 과제에 적용할 수 있을까

NICEST의 아이디어는 다른 비전 및 언어 이해 과제에도 적용할 수 있습니다. 예를 들어, 이미지 캡션 생성이나 시각적 질문 응답과 같은 작업에서도 노이즈 레이블 문제가 발생할 수 있습니다. NICEST의 노이즈 레이블 수정 및 훈련 전략은 이러한 작업에서도 모델의 성능을 향상시키고 더 강건한 학습을 가능하게 할 수 있습니다. 또한, 다중 교사 지식 증류를 활용하여 모델을 훈련하는 방법은 다양한 비전 및 언어 이해 작업에서 모델의 일반화 능력을 향상시키는 데 유용할 수 있습니다.