期刊
ADVANCED INFORMATION NETWORKING AND APPLICATIONS, AINA-2022, VOL 1
卷 449, 期 -, 页码 375-386出版社
SPRINGER INTERNATIONAL PUBLISHING AG
DOI: 10.1007/978-3-030-99584-3_33
关键词
Image segmentation; Federated learning; Intravascular ultrasound
资金
- Ministry of Science and Technology (MOST), Taiwan [MOST 110-2222-E-001-002, 1102221-E-002-078-MY2, 110-2321-B-075-002]
A federated learning-based deep learning model is proposed for distributed IVUS image segmentation. The results indicate that the lightweight model performs almost as well as deeper models but with significantly less time consumption.
Deep learning technologies have been widely used in intravascular ultrasound (IVUS) image analysis. An efficient and real-time tool for IVUS image segmentation is essential during percutaneous coronary intervention (PCI). However, collecting a significant volume of data is infeasible lying to the following reasons: (1) patients' privacy concerns and (2) manually annotating data is time-consuming and laborious. Therefore, training a deep model with generalization capacity using distributed data remains a challenge. In this study, a federated learning-based deep learning model is proposed. It is a framework that enables model training on multi-source data without sharing each local dataset. A lightweight U-Net model is also adopted in the federated learning framework due to the limited volume of data. The results indicate that the lightweight model's performance can almost achieve the performance of a deeper model but significantly spends less time consumption. The proposed lightweight U-Net model parameters are less than 30 times that of the baseline model. The external elastic membrane (EEM) and the lumen boundary segmentation achieved 0.8567 and 0.8457, respectively. The efficiency of implementing such a federated learning framework on a distributed IVUS image segmentation is estimated to be increased by more than four times.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据