It Takes Two: Complementary Self-Distillation for Contextual Integrity in LLMs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Park, Sangwoo, Yeo, Woongyeong, Lee, Seanie, Choi, Yumin, Lee, Hyomin, Kim, Kangsan, Baek, Jinheon, Oh, Seong Joon, Hwang, Sung Ju
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913146644463616
author Park, Sangwoo
Yeo, Woongyeong
Lee, Seanie
Choi, Yumin
Lee, Hyomin
Kim, Kangsan
Baek, Jinheon
Oh, Seong Joon
Hwang, Sung Ju
author_facet Park, Sangwoo
Yeo, Woongyeong
Lee, Seanie
Choi, Yumin
Lee, Hyomin
Kim, Kangsan
Baek, Jinheon
Oh, Seong Joon
Hwang, Sung Ju
contents Contextual Integrity (CI) defines privacy not merely as keeping information hidden, but as governing information flows according to the norms of a given context. As large language models are increasingly deployed as personal agents handling sensitive workflows, adhering to CI becomes critical. However, even frontier models remain unreliable in making disclosure decisions, and existing mitigation strategies often degrade underlying task performance. To overcome this privacy-utility trade-off, we propose SELFCI, a complementary self-distillation framework that decouples information suppression from task resolution. SELFCI jointly optimizes two independent reverse KL divergences over distinct teacher distributions derived from feedback: one encourages preserving task-relevant information for utility, while the other enforces minimal and appropriate disclosure. This complementary formulation induces a Product-of-Experts (PoE) target, aligning the policy with the intersection of capability and privacy requirements. Empirical evaluations demonstrate that SELFCI, without relying on costly external supervision, consistently outperforms competitive baselines such as online reinforcement learning algorithms (e.g., GRPO). These trends further extend to out-of-domain settings involving agentic workflows and accumulated private context, suggesting that SELFCI provides a practical path toward CI alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2605_20258
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle It Takes Two: Complementary Self-Distillation for Contextual Integrity in LLMs
Park, Sangwoo
Yeo, Woongyeong
Lee, Seanie
Choi, Yumin
Lee, Hyomin
Kim, Kangsan
Baek, Jinheon
Oh, Seong Joon
Hwang, Sung Ju
Machine Learning
Artificial Intelligence
Cryptography and Security
Contextual Integrity (CI) defines privacy not merely as keeping information hidden, but as governing information flows according to the norms of a given context. As large language models are increasingly deployed as personal agents handling sensitive workflows, adhering to CI becomes critical. However, even frontier models remain unreliable in making disclosure decisions, and existing mitigation strategies often degrade underlying task performance. To overcome this privacy-utility trade-off, we propose SELFCI, a complementary self-distillation framework that decouples information suppression from task resolution. SELFCI jointly optimizes two independent reverse KL divergences over distinct teacher distributions derived from feedback: one encourages preserving task-relevant information for utility, while the other enforces minimal and appropriate disclosure. This complementary formulation induces a Product-of-Experts (PoE) target, aligning the policy with the intersection of capability and privacy requirements. Empirical evaluations demonstrate that SELFCI, without relying on costly external supervision, consistently outperforms competitive baselines such as online reinforcement learning algorithms (e.g., GRPO). These trends further extend to out-of-domain settings involving agentic workflows and accumulated private context, suggesting that SELFCI provides a practical path toward CI alignment.
title It Takes Two: Complementary Self-Distillation for Contextual Integrity in LLMs
topic Machine Learning
Artificial Intelligence
Cryptography and Security
url https://arxiv.org/abs/2605.20258