提出黑箱优化方法BBoxER,实现大模型后训练的隐私保护与强泛化保障
Tuning without Peeking: Provable Generalization Bounds and Robust LLM Post-Training
- 通过隐式压缩数据构建信息瓶颈,仅依赖函数值优化
- 在推理数据集上性能提升,且对成员推断攻击鲁棒
- 首次提供非平凡泛化界,适合高安全性场景
梯度优化虽高效可扩展,但暴露梯度可能泄露敏感数据,易受数据投毒等攻击。本文提出BBoxER,一种基于进化的黑箱优化方法,通过隐式压缩训练数据构建信息瓶颈。利用信息流的可处理性,给出非平凡的泛化界及对数据投毒和数据提取攻击的强理论保障,同时实现设计即隐私。在大模型实验中,仅需少数迭代,BBoxER即能提升性能,良好泛化于推理基准数据集,并抵御成员推断攻击。该方法作为梯度优化的补充,适用于受限环境,兼具理论保证与实用性。
原文摘要 · Abstract (English)
Gradient-based optimization is the workhorse of deep learning, offering efficient and scalable training via backpropagation. However, exposing gradients during training can leak sensitive information about the underlying data, raising privacy and security concerns such as susceptibility to data poisoning attacks. In contrast, black-box optimization methods, which treat the model as an opaque function, relying solely on function evaluations to guide optimization, offer a promising alternative in scenarios where data access is restricted, adversarial risks are high, or overfitting is a concern. This paper introduces BBoxER, an evolutionary black-box method for LLM post-training that induces an information bottleneck via implicit compression of the training data. Leveraging the tractability of information flow, we provide non-vacuous generalization bounds and strong theoretical guarantees for robustness to data poisoning attacks and extraction attacks, while ensuring privacy by design. In experiments with LLMs, we demonstrate empirically that black-box optimization methods-despite the scalability and computational challenges inherent to black-box approaches-are able to learn, showing how a few iterations of BBoxER improve performance, generalize well on a benchmark of reasoning datasets, and are robust to membership inference attacks. This positions BBoxER as an attractive add-on on top of gradient-based optimization, offering suitability for deployment in restricted environments while also providing non-vacuous generalization guarantees.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。