RoGA: Towards Generalizable Deepfake Detection through Robust Gradient Alignment
Created by
Haebom
저자
Lingyu Qiu, Ke Jiang, Xiaoyang Tan
개요
본 논문은 딥페이크 탐지에서 도메인 일반화를 위한 새로운 학습 목표를 제시합니다. 기존 방법들은 과적합을 방지하기 위해 추가 모듈을 사용하지만, 이는 경험적 위험 최소화(ERM) 목표의 최적화를 저해하여 성능 저하를 초래할 수 있습니다. 본 논문에서는 일반화 기울기 업데이트와 ERM 기울기 업데이트를 정렬하는 새로운 학습 목표를 제안합니다. 모델 파라미터에 대한 섭동을 적용하여 도메인 간 상승점을 정렬함으로써 딥페이크 탐지 모델의 도메인 변화에 대한 강건성을 향상시킵니다. 추가적인 규제 없이 도메인 불변 특징을 유지하면서 도메인 특정 특징을 관리합니다. 다양한 딥페이크 탐지 데이터셋에 대한 실험 결과는 제안된 기울기 정렬 전략이 최첨단 도메인 일반화 기법들을 능가함을 보여줍니다. 소스 코드는 https://github.com/Lynn0925/RoGA 에서 이용 가능합니다.