End-to-end Learning of Sparse Interventions on Activations to Steer Generation
Created by
Haebom
Category
Empty
저자
Pau Rodriguez, Michal Klein, Eleonora Gualdoni, Arno Blaas, Luca Zappella, Marco Cuturi, Xavier Suau
개요
본 논문은 생성 모델의 출력을 제어하기 위한 효율적인 메커니즘으로서 선형 엔드-투-엔드 활성화 조향(LinEAS)을 제안합니다. 기존의 활성화 조작 기반 방법들이 국소적인 조정으로 인해 예측 불가능한 결과를 초래하는 한계를 극복하기 위해, LinEAS는 모든 계층의 분포 차이를 동시에 고려하는 전역 손실 함수를 사용하여 학습됩니다. 또한, 희소성 규제를 통해 뉴런 및 계층 선택을 자동으로 수행하여 효율성을 높입니다. 실험 결과, LinEAS는 적은 수의 샘플만으로도 유해성 완화 등에서 기존 방법보다 우수한 성능을 보이며, 훨씬 복잡한 미세 조정 방식과 유사한 성능을 달성합니다. 텍스트-이미지 확산 모델에도 적용 가능성을 보여줍니다.