IEEE J Biomed Health Inform. 2023 Nov;27(11):5249-5259. doi: 10.1109/JBHI.2023.3247463. Epub 2023 Nov 7.
The Healthcare Internet-of-Things (IoT) framework aims to provide personalized medical services with edge devices. Due to the inevitable data sparsity on an individual device, cross-device collaboration is introduced to enhance the power of distributed artificial intelligence. Conventional collaborative learning protocols (e.g., sharing model parameters or gradients) strictly require the homogeneity of all participant models. However, real-life end devices have various hardware configurations (e.g., compute resources), leading to heterogeneous on-device models with different architectures. Moreover, clients (i.e., end devices) may participate in the collaborative learning process at different times. In this paper, we propose a Similarity-Quality-based Messenger Distillation (SQMD) framework for heterogeneous asynchronous on-device healthcare analytics. By introducing a preloaded reference dataset, SQMD enables all participant devices to distill knowledge from peers via messengers (i.e., the soft labels of the reference dataset generated by clients) without assuming the same model architecture. Furthermore, the messengers also carry important auxiliary information to calculate the similarity between clients and evaluate the quality of each client model, based on which the central server creates and maintains a dynamic collaboration graph (communication graph) to improve the personalization and reliability of SQMD under asynchronous conditions. Extensive experiments on three real-life datasets show that SQMD achieves superior performance.
医疗保健物联网 (IoT) 框架旨在利用边缘设备提供个性化医疗服务。由于单个设备上不可避免的数据稀疏性,引入了跨设备协作来增强分布式人工智能的能力。传统的协作学习协议(例如,共享模型参数或梯度)严格要求所有参与模型的同质性。然而,现实生活中的终端设备具有各种硬件配置(例如,计算资源),导致具有不同架构的设备上模型异构。此外,客户端(即终端设备)可能在不同的时间参与协作学习过程。在本文中,我们提出了一种用于异构异步设备端医疗保健分析的基于相似度-质量的信使蒸馏 (SQMD) 框架。通过引入预加载的参考数据集,SQMD 允许所有参与设备通过信使(即客户端生成的参考数据集的软标签)从同行中提取知识,而无需假设相同的模型架构。此外,信使还携带重要的辅助信息来计算客户端之间的相似度并评估每个客户端模型的质量,基于此,中央服务器创建和维护一个动态协作图(通信图),以提高异步条件下 SQMD 的个性化和可靠性。在三个真实数据集上的广泛实验表明,SQMD 实现了卓越的性能。