不共享数据也能提升模型性能,用缝合技术融合独立训练的模型
Sharing Knowledge without Sharing Data: Stitches can improve ensembles of disjointly trained models
- 通过缝合层整合独立训练模型的中间特征
- 在各自数据上性能恢复至接近单模型水平
- 适合数据无法共享但需协作的医疗等领域
深度学习在诸多现实任务中表现优异,但依赖大规模多样化数据。在医疗等场景中,数据分散且难以共享。联邦学习虽能解决此问题,但要求各方同步训练并交换权重。本文研究异步协作——仅共享已训练模型(如论文发布时)——对性能的影响,提出使用缝合技术融合模型。从多方独立视角出发,发现仅在本方数据上训练的模型,在合并后仍可保持相近性能,但在其他方数据上表现较差。虽然此类模型的集成能提升泛化能力,但会损害各自主数据上的表现。通过在独立训练模型的中间表示中引入精心设计的缝合层,可使性能恢复至具有竞争力的水平,同时保持更好泛化性,证明异步协作亦可获得良好结果。
原文摘要 · Abstract (English)
Deep learning has been shown to be very capable at performing many real-world tasks. However, this performance is often dependent on the presence of large and varied datasets. In some settings, like in the medical domain, data is often fragmented across parties, and cannot be readily shared. While federated learning addresses this situation, it is a solution that requires synchronicity of parties training a single model together, exchanging information about model weights. We investigate how asynchronous collaboration, where only already trained models are shared (e.g. as part of a publication), affects performance, and propose to use stitching as a method for combining models. Through taking a multi-objective perspective, where performance on each parties' data is viewed independently, we find that training solely on a single parties' data results in similar performance when merging with another parties' data, when considering performance on that single parties' data, while performance on other parties' data is notably worse. Moreover, while an ensemble of such individually trained networks generalizes better, performance on each parties' own dataset suffers. We find that combining intermediate representations in individually trained models with a well placed pair of stitching layers allows this performance to recover to a competitive degree while maintaining improved generalization, showing that asynchronous collaboration can yield competitive results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。