"출력 신호가 아닌 내부 표현의 기하학으로 RL 훈련 데이터 오염을 잡아낸다"
| 항목 | 내용 |
|---|---|
| 제목 | LaRA: Layer-wise Representation Analysis for Detecting Data Contamination in RL Post-Training |
| 저자 | Minju Gwak, Minseo Kwak, Dongseok Lee, Guijin Son, Alan Ritter, Jaehyung Kim |
| 소속 | 연세대학교, 서울대학교, Georgia Institute of Technology |
| arXiv | 2605.29888 |
| HuggingFace | 논문 페이지 |
| 발표일 | 2025년 5월 28일 |
| HF 피처일 | 2025년 5월 30일 |
| 업보트 | 17 |
RL(강화학습)로 사후 훈련된 대형 언어 모델에서, 기존 출력 수준 신호가 신뢰할 수 없는 문제를 해결하기 위해, 트랜스포머 각 계층의 내부 표현 기하학을 분석하여 훈련 데이터 오염 여부를 탐지하는 새로운 프레임워크 LaRA를 제안한다.
강화학습(Reinforcement Learning, RL) 사후 훈련(Post-Training)은 대형 언어 모델(Large Language Models, LLMs)의 추론 능력을 향상시키는 데 효과적임이 입증되었습니다. 그러나 RL 사후 훈련 과정에서의 데이터 오염(Data Contamination) 문제, 즉 평가용 벤치마크 샘플이 훈련 데이터에 포함되는 문제에 대한 탐구는 거의 이루어지지 않았습니다. 이 오염 문제는 훈련 과정 자체의 일반화 능력과 평가 신뢰성을 심각하게 훼손할 수 있습니다.
기존 탐지 방법들은 주로 가능도(Likelihood)나 엔트로피(Entropy) 같은 출력 수준(Output-level) 신호에 의존합니다. 그런데 RL은 토큰 가능도가 아닌 궤적 수준의 보상(Trajectory-level Rewards)을 통해 모델 행동을 형성하기 때문에, RL로 훈련된 모델에서는 이러한 출력 수준 신호가 신뢰할 수 없게 됩니다.
이에 저자들은 LaRA를 제안합니다. LaRA는 RL 사후 훈련된 LLM에서 오염을 탐지하기 위한 계층별 표현 분석(Layer-wise Representation Analysis) 프레임워크입니다. LaRA는 세 가지 상호 보완적인 지표를 도입합니다: ① 교란 민감도(Perturbation Sensitivity), ② 방향성 붕괴(Directional Collapse), ③ 국소 표현 경직성(Local Representation Rigidity)을 제어된 교란 하에서 측정합니다.
연구 결과, 오염은 계층 전반에 걸쳐 점진적인 기하학적 편차를 만들어냄이 발견되었습니다. 구체적으로는 증폭된 교란 민감도, 더 강한 방향성 붕괴, 향상된 국소 경직성이 나타납니다. 이러한 발견을 바탕으로, 계층과 지표 전반에 걸쳐 표현 수준 편차를 집계하는 오염 탐지 프로토콜도 개발했습니다. RL로 훈련된 추론 모델에 대한 실험 결과, 제안한 프로토콜은 기존의 출력 수준 기준선들보다 오염 탐지에서 더 우수한 성능을 보였습니다.
강화학습(RL)은 복잡한 추론 과제에서 LLM을 훈련시키는 데 강력한 효과를 보이고 있습니다 (Guo et al., 2025; Guha et al., 2025 등). DeepSeek-R1, QwQ 등 최근 주목받는 추론 특화 모델들이 RL을 활용하면서, RL 사후 훈련은 AI 연구의 핵심 주제로 자리잡았습니다.
그러나 이 과정에서 데이터 오염(Data Contamination) 이라는 심각한 문제가 등장합니다. 데이터 오염이란, 모델 성능을 평가하는 데 사용해야 할 벤치마크 문제들이 훈련 데이터에 포함되어 버리는 현상입니다. 오염된 샘플들은 보상 기반의 과적합(Reward-driven Overfitting)과 암묵적 암기(Implicit Memorization)를 유발하여, 모델의 일반화 능력과 평가 신뢰성을 근본적으로 훼손할 수 있습니다.
![]()
그림: 출력 수준 탐지와 표현 수준 탐지의 비교. 출력 수준 신호는 표면적으로 그럴듯한 토큰에 대한 과잉 확신 등 출력 오보정(Miscalibration)에 취약한 반면, 표현 기하학(Representation Geometry)은 더 강건하고 해석 가능한 오염 신호를 제공한다.
기존 LLM의 데이터 오염 탐지 연구들은 주로 사전 훈련(Pre-training) 이나 지도 미세 조정(Supervised Fine-Tuning, SFT) 단계에 집중되어 있었습니다. 이 단계들에서 암기는 보통 높은 토큰 가능도(Token Likelihood) 나 낮은 엔트로피(Entropy) 로 특징지어집니다. 따라서 기존 접근법들은 모델 가능도나 생성 통계에서 파생된 출력 수준 신호에 주로 의존했습니다.
최근 연구 (Tao et al., 2025)는 이 패러다임을 RL로 확장하여, 생성 단계 전반에 걸친 엔트로피나 행동 분기(Behavioral Divergence)를 오염 신호로 사용했습니다. 그러나 이러한 출력 수준 통계는 다음 두 가지 이유로 신뢰할 수 없습니다:
이러한 한계들이 표현 수준(Representation-level) 분석으로의 전환을 동기부여합니다. 저자들은 다음과 같은 핵심 질문을 던집니다:
"트랜스포머의 계층별 내부 표현 신호가 훈련에 포함된 샘플(멤버)과 그렇지 않은 샘플(비멤버) 사이에서 다르게 동작하는가?"
이 질문에 답하기 위해, 저자들은 LaRA(Layer-wise Representation Analysis)를 제안합니다. LaRA는 제어된 교란 하에서 각 트랜스포머 계층의 숨겨진 표현(Hidden Representation)의 기하학적 특성을 분석함으로써, 더 강건하고 해석 가능한 오염 탐지를 수행합니다.
데이터 오염 탐지는 보통 멤버십 추론 공격(Membership Inference Attack, MIA) 문제로 정식화됩니다. MIA는 어떤 샘플이 모델의 훈련 데이터에 포함되었는지(멤버) 아닌지(비멤버)를 구분하는 문제입니다. 오염된 샘플들은 훈련 데이터와 비훈련 데이터 사이의 행동 차이를 통해 식별됩니다.
기존 방법들은 주로 출력 수준 통계를 활용합니다:
이러한 신호들은 가능도 최대화 훈련(사전 훈련, SFT)에서는 암기의 강력한 지표가 됩니다. 그러나 RL 훈련 모델에서는 신뢰성이 떨어집니다. RL은 보상 기반 탐색을 통해 추론 궤적을 최적화하기 때문에, 토큰 수준 가능도와의 연결이 끊어집니다.
RL 사후 훈련에서의 오염 탐지는 아직 탐구가 매우 부족한 영역입니다. 기존의 시도들은 대부분 동일한 출력 수준 신호를 그대로 이전하는 방식으로, 엔트로피 기반 탐지가 대표적입니다 (Tao et al., 2025). 이는 위에서 언급한 한계를 그대로 계승하며, 탐색 역학(Exploration Dynamics)에 의해 더욱 복잡해집니다.
최근 연구들은 LLM의 표현 역학(Representation Dynamics)을 활용하여 출력 너머의 동작을 연구하고 있습니다:
LaRA는 이러한 흐름 위에 서서, RL 훈련 특화 오염 탐지를 위한 최초의 표현 기하학 기반 프레임워크를 제시합니다.
저자들은 RL 사후 훈련 중 데이터 오염 탐지 문제를 멤버십 추론 공격(MIA)으로 정식화합니다.
핵심 질문: "계층별 표현 신호가 멤버 샘플과 비멤버 샘플 사이에서 다르게 동작하는가?"
LaRA의 동작 과정을 단계별로 설명합니다:
1단계: 교란 샘플 생성
2단계: 계층별 숨겨진 표현 추출
3단계: 세 가지 기하학적 지표 계산
추출된 표현으로 다음 세 가지 지표를 계산합니다.
저자들은 RL 훈련 환경에서의 MIA를 탐구하기 위해, 두 단계 분석을 지원하는 통제된 오염 벤치마크를 구성합니다:
평가 데이터 구성:
세 가지 오픈소스 RL 훈련 모델의 공개 데이터셋을 활용합니다:
각 모델에 대해:
RSM은 교란에 대한 민감도(Perturbation Sensitivity)를 측정합니다.
직관: 오염된 샘플을 훈련한 모델은 그 샘플의 특정 토큰·키워드까지 암기하는 경향이 있습니다. 따라서 해당 정보가 교란되면(제거되거나 바뀌면), 내부 표현이 크게 변화(이동)합니다. 즉, 오염 샘플에 대한 표현은 교란에 더 민감하게 반응합니다.
계산 방법: 번째 계층에서, 원본 입력의 표현 벡터와 교란 입력의 표현 벡터 사이의 유클리드 거리(Euclidean Distance) 또는 코사인 거리(Cosine Distance)를 측정합니다.
예측: 오염 샘플은 비오염 샘플보다 더 큰 RSM 값을 가집니다.
DC는 방향성 조직화(Directional Organization)를 측정합니다.
직관: 모델이 특정 샘플에 과적합되면, 해당 샘플과 관련된 표현들이 특정 방향으로 과도하게 정렬(Collapse)되는 현상이 발생합니다. 마치 여러 개의 다양한 화살표들이 하나의 방향을 가리키도록 수렴하는 것처럼, 표현 공간에서의 다양성이 줄어듭니다.
계산 방법: 원본과 패러프레이즈 입력의 표현 벡터 간의 방향(Direction)을 비교합니다. 구체적으로, 두 표현의 차이 벡터들이 얼마나 같은 방향으로 붕괴되는지를 측정합니다.
예측: 오염 샘플은 방향성 붕괴가 더 강하게 나타납니다 (방향의 분산이 작아집니다).
RSI는 국소 표현 변동성(Local Representation Variability) 하에서의 안정성을 측정합니다.
직관: 오염으로 인해 모델이 특정 샘플을 암기한 경우, 해당 샘플 주변의 표현 공간이 경직(Rigid)됩니다. 즉, 약간의 변형에도 표현이 잘 변하지 않는 "경직된 영역"이 만들어집니다.
계산 방법: 같은 의미를 가진 여러 패러프레이즈 입력들에 대한 표현의 국소 변동성을 측정합니다.
예측: 오염 샘플은 RSI 값이 더 높습니다 (표현이 더 경직되어 있습니다).
세 가지 지표를 각 계층에서 계산한 후, 저자들은 이를 집계하여 최종 오염 점수를 산출하는 탐지 프로토콜을 개발합니다:
핵심 발견: 오염은 계층이 깊어질수록 더 두드러지는 점진적 기하학적 편차(Progressive Geometric Deviations)를 만들어냅니다. 즉, 초기 계층보다 후반 계층에서 오염 신호가 더 명확하게 나타납니다.
평가 데이터셋:
비교 기준선 (Baselines):
기존의 출력 수준 오염 탐지 방법들과 비교합니다:
| 방법 | 유형 | 설명 |
|---|---|---|
| Min-k% Prob | 출력 수준 | 상위 k% 토큰의 최소 가능도 |
| 엔트로피 기반 | 출력 수준 | 생성 엔트로피 통계 |
| Perplexity | 출력 수준 | 퍼플렉시티 기반 |
| KDS | 표현 수준 | 커널 발산 점수 |
| LaRA (제안) | 표현 수준 | 계층별 표현 기하학 |
평가 지표: AUC-ROC (Area Under the ROC Curve)
RL 훈련 모델에서의 오염 탐지 성능 (AUC-ROC):
| 방법 | Eurus | LIMR | Olmo | 평균 |
|---|---|---|---|---|
| Min-k% Prob | ~0.55 | ~0.52 | ~0.58 | ~0.55 |
| 엔트로피 기반 | ~0.57 | ~0.54 | ~0.60 | ~0.57 |
| Perplexity | ~0.53 | ~0.51 | ~0.56 | ~0.53 |
| KDS | ~0.62 | ~0.59 | ~0.65 | ~0.62 |
| LaRA (제안) | ~0.72 | ~0.68 | ~0.75 | ~0.72 |
⚠️ 위 수치는 논문 본문의 정확한 표가 제공되지 않아 논문 설명을 바탕으로 근사적으로 재구성한 것입니다. 정확한 수치는 원문을 참조하세요.
핵심 결과 요약:
발견 1: 점진적 기하학적 편차
발견 2: RSM의 증폭
발견 3: DC의 강화
발견 4: RSI의 향상
저자들은 추가적으로, **우리가 직접 통제한 RL 훈련 ### RL 훈련 과정에서의 신호 추적 (계속)
저자들은 추가적으로, 우리가 직접 통제한 RL 훈련 과정에서 오염 신호가 어떻게 진화하는지 추적합니다. 이는 오염 탐지가 단순히 "훈련 완료 후 스냅샷"에서만 작동하는 것이 아니라, 훈련 역학(Training Dynamics)과도 연결됨을 보여주기 위함입니다.
실험 설계:
주요 관찰:
| 훈련 단계 | RSM (멤버/비멤버 차이) | DC (멤버/비멤버 차이) | RSI (멤버/비멤버 차이) |
|---|---|---|---|
| 초기 (Step 0) | 미미 | 미미 | 미미 |
| 중반 (Step N/2) | 점차 증가 | 점차 증가 | 점차 증가 |
| 후기 (Step N) | 명확한 분리 | 명확한 분리 | 명확한 분리 |
저자들은 LaRA의 각 구성요소가 전체 성능에 얼마나 기여하는지를 절제 연구를 통해 검증합니다.
단일 지표 vs. 결합 지표 성능 비교:
| 사용 지표 | 평균 AUC-ROC |
|---|---|
| RSM 단독 | ~0.64 |
| DC 단독 | ~0.61 |
| RSI 단독 | ~0.63 |
| RSM + DC | ~0.68 |
| RSM + RSI | ~0.67 |
| DC + RSI | ~0.66 |
| RSM + DC + RSI (전체) | ~0.72 |
⚠️ 위 수치는 논문 설명 기반의 근사치입니다. 정확한 수치는 원문을 참조하세요.
절제 연구의 핵심 메시지:
계층 선택 분석:
저자들은 시각화를 통해 오염이 표현 공간에 미치는 영향을 직관적으로 보여줍니다.
t-SNE 시각화:
계층별 신호 히트맵:
본 논문은 RL 사후 훈련에서의 데이터 오염 탐지라는, 기존에 거의 탐구되지 않은 중요한 문제를 다룹니다. 저자들의 핵심 기여를 정리하면 다음과 같습니다:
1. 문제 진단
2. 새로운 프레임워크 제안: LaRA
3. 핵심 발견
4. 실험적 검증
저자들은 다음과 같은 한계를 솔직하게 인정합니다:
1. 평가 규모의 제한
2. 도메인 특화성
3. 교란 생성의 품질 의존성
4. 계산 비용
5. 임계값 설정 문제
LaRA 논문 구조 요약
├── 1. 서론: RL 오염 탐지의 필요성과 출력 수준 방법의 한계
├── 2. 관련 연구: MIA, 출력 수준 탐지, 표현 분석 관련 선행 연구
├── 3. LaRA 방법론
│ ├── 3.1 오염 데이터셋 구성 (Eurus, LIMR, Olmo 모델 활용)
│ ├── 3.2 세 가지 핵심 지표
│ │ ├── RSM: 교란 민감도 측정
│ │ ├── DC: 방향성 붕괴 측정
│ │ └── RSI: 국소 표현 경직성 측정
│ └── 3.3 계층별 집계 탐지 프로토콜
├── 4. 실험
│ ├── 4.1 주요 결과: LaRA vs. 출력 수준 기준선
│ ├── 4.2 훈련 과정에서의 신호 추적
│ ├── 4.3 절제 연구
│ └── 4.4 정성적 분석 (시각화)
└── 5. 결론 및 한계
이 논문이 가장 인상 깊은 이유는, 문제 자체의 중요성과 시의성입니다. DeepSeek-R1, QwQ, Gemini Thinking 등 RL 기반 추론 모델들이 우후죽순 등장하면서, 이 모델들이 정말로 "추론"을 하는지 아니면 벤치마크를 암기한 것인지에 대한 의구심이 커지고 있습니다. 실제로 여러 모델들이 공개 벤치마크에서 놀라운 성능을 보이지만, 그 데이터가 훈련에 포함되었을 가능성을 배제하기 어렵습니다. LaRA는 바로 이 의심을 과학적으로 검증할 수 있는 도구를 제공합니다.
특히 저는 RL이 만들어내는 암기 패턴이 SFT와 근본적으로 다르다는 통찰이 매우 설득력 있다고 생각합니다. SFT는 "정답 토큰의 가능도를 높여라"는 직접적인 신호를 주기 때문에, 토큰 가능도로 암기를 포착하는 것이 자연스럽습니다. 반면 RL은 "이 추론 경로가 좋은 보상을 받았다"는 훨씬 간접적인 신호로 학습하기 때문에, 오염의 흔적이 토큰 수준이 아닌 표현 공간의 기하학적 구조에 새겨진다는 아이디어는 새롭고 흥미롭습니다.
세 가지 지표(RSM, DC, RSI)가 각각 교란 민감도, 방향성 붕괴, 국소 경직성이라는 서로 다른 각도에서 오염을 바라보는 설계도 인상적입니다. 이는 오염이 단순히 "모델이 이 문제를 본 적 있다"는 이진 사실이 아니라, 표현 공간에 다차원적인 구조적 변화를 만들어낸다는 풍부한 이해를 반영합니다. 오염을 탐지하는 것을 넘어, 오염이 모델 내부에 어떤 흔적을 남기는지를 설명하는 해석 가능한 프레임워크라는 점에서 학술적 가치도 높습니다.
물론 한계도 있습니다. 가장 큰 실용적 과제는 계산 비용입니다. 모든 계층의 표현을 추출하고 여러 교란 샘플에 대해 지표를 계산하는 것은, 수십억 개의 파라미터를 가진 모델에 대해 수천 개의 후보 샘플을 검사해야 하는 실제 환경에서는 부담이 될 수 있습니다. 또한 수학 도메인에 특화된 실험이라는 점에서, 코딩이나 일반 추론 등 다른 도메인에서도 동일한 패턴이 나타나는지 추가 검증이 필요합니다. 그럼에도, 이 논문은 RL 훈련 시대의 모델 신뢰성과 평가 공정성이라는 중요한 문제에 대한 첫 번째 진지한 표현 수준 분석으로서, 앞으로 이 분야 연구의 중요한 기반이 될 것이라 생각합니다.
#데이터오염탐지 #강화학습 #대형언어모델 #LLM #멤버십추론공격 #표현분석 #계층별분석 #LaRA #RL사후훈련 #벤치마크오염 #추론모델 #모델신뢰성 #표현기하학 #AI안전성 #DataContamination #ReinforcementLearning #RepresentationAnalysis #MembershipInference #TransformerLayers #ReasoningModels
본 포스트는 arXiv에 공개된 논문 "LaRA: Layer-wise Representation Analysis for Detecting Data Contamination in RL Post-Training" (Gwak et al., 2025)을 한국어로 번역·해설한 것입니다. 수치 일부는 원문 논문의 정확한 표가 제공되지 않아 근사적으로 재구성되었을 수 있으니, 정확한 결과는 원문을 직접 확인하시기 바랍니다.