Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

GUARD: Guided Unlearning and Retention via Data Attribution for Large Language Models

Created by
  • Haebom

저자

Peizhi Niu, Evelyn Ma, Huiting Zhou, Duo Zhou, Huan Zhang, S. Rasoul Etesami, Olgica Milenkovic

개요

본 논문은 규제 준수, 저작권 보호 및 개인 정보 보호 문제로 인해 대규모 언어 모델(LLM)에서 언러닝의 중요성이 증가함에 따라, 의도하지 않은 망각 문제 해결에 초점을 맞춘다. 특히, 특정 데이터 제거가 모델의 유용성을 저해하는 문제를 해결하기 위해 데이터 수준 요인이 언러닝 성능에 미치는 영향을 연구한다. GUARD라는 새로운 프레임워크를 제안하며, 이는 데이터 속성을 기반으로 한 가중 언러닝을 통해 고충격 데이터의 유지를 향상시킨다. GUARD는 계산 효율적인 프록시 데이터 속성 메트릭을 도입하여 Forget 및 Retain 세트 간의 정렬을 정량화하고, 속성 점수에 반비례하는 가변적인 언러닝 가중치를 샘플에 할당하는 새로운 언러닝 목표를 설계한다.

시사점, 한계점

시사점:
GUARD는 LLM 언러닝에서 의도하지 않은 유지 손실을 완화한다.
데이터 속성을 활용한 가중 언러닝 방식을 통해 TOFU 및 MUSE 벤치마크에서 유용성 희생을 줄이고 지식 유지를 향상시켰다.
기존 방법과 비교하여 향상된 유지 성능을 보이면서도, 프라이버시 손실은 유사하거나 소폭 증가하는 수준을 유지한다.
이론적 보장을 통해 GUARD의 성능 우수성을 뒷받침한다.
한계점:
데이터 속성 메트릭의 효율성은 입증되었지만, 실제 데이터 환경에서의 일반화 가능성에 대한 추가 연구가 필요할 수 있다.
특정 LLM 아키텍처 및 벤치마크에 대한 실험 결과를 제시하고 있으므로, 다양한 모델 및 데이터셋에서의 성능 검증이 필요하다.
프라이버시 손실의 증가가 미미하다고 언급되었지만, 실제 환경에서 적용 시 프라이버시 관련 추가적인 고려가 필요할 수 있다.
👍