返回
FedRNL: Federated Rationalization with Soft Parameter Sharing
DOI:10.1007/978-3-032-05981-9_21.png)
摘要
En 中文
可解释性在自然语言处理中至关重要,能够增强透明度和信任度。理性化模型通过提取关键输入片段(即理性依据)来解释决策,同时保持预测性能。另一方面,联邦学习(FL)近年来已成为训练机器学习模型的关键范式,因为它可以利用来自多个客户端的训练数据,而无需上传其原始数据。基于此,我们首先提出以联邦学习的方式训练理性化模型。然而,我们发现简单地将两者结合会导致严重性能下降,这是由于客户端之间的数据异构性,其中存在不一致的理性依据生成。为解决此问题,我们提出FedRNL,该模型引入软共享机制来对齐生成器和预测器的编码器,确保浅层一致性和深层泛化。一个编码器损失最小化特征差异,而分层聚合策略分别更新服务器上的生成器和预测器,增强模型稳定性。大量实验表明,FedRNL相较于现有的通用异构性缓解方法显著提升了性能。
Keyword:
Federated learning
Rationalization
Non-IID
期刊
机构
引用论文
Learning Phrase Representations using RNN Encoder–Decoder for Statistical Machine Translation使用RNN编码器-解码器学习统计机器翻译的短语表示

