Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2210003(2023)

Lung Nodule CT Image Classification Based on Adaptive Aggregate Weight Federated Learning

Jiangfeng Shi1, Bao Feng2、*, Yehang Chen2, and Xiangmeng Chen3
Author Affiliations
  • 1School of Electronic Engineering and Automation, Guilin University of Electronic Technology, Guilin 541004, Guangxi , China
  • 2Laboratory of Artificial Intelligence of Biomedicine, Guilin University of Aerospace Technology, Guilin 541004, Guangxi , China
  • 3Laboratory of Intelligent Computing and Application of Medical Imaging, Jiangmen Central Hospital, Jiangmen 529030, Guangdong , China
  • show less

    The field of medical imaging currently faces the problems of data island and non-independent and independently distributed (Non-IID) variables in multi-center data. In this study, a federated learning algorithm based on adaptive aggregate weight (FedAaw) is proposed. Using a global model polymerization process, this study utilized the accuracy threshold to filter out the local model; the model accuracy is calculated by the center server. The corresponding weights of polymerization, which are updated in the global model, yielded models with better classification performances that are used to construct a global model, which helps address the problems associated with Non-IID multicenter data. Furthermore, to improve the applicability of the model to mining the information between the long and short distance of the image, the multi head self-attention mechanism is introduced to the local and global models. In addition, to address the problem of model overfitting caused by end-to-end redundant features, the convolution kernel features in the global model are extracted. The learning of sparse Bayesian extreme learning machine based on L1 norm (SBELML1) framework is used for the feature classification of the data obtained from each center. Finally, the anti-interference ability of the FedAaw algorithm is verified by shuffling the data distribution of different centers several times. The AUC ranges of the test sets used in the five centers are as follows: center 1: (0.7947?0.8037), center 2: (0.8105?0.8405), center 3: (0.6768?0.7758), center 4: (0.8496?0.9063), and center 5: (0.8913?0.9348). These results indicate that FedAaw has good classification performance on multi-center data and a strong anti-interference ability.

    Tools

    Get Citation

    Copy Citation Text

    Jiangfeng Shi, Bao Feng, Yehang Chen, Xiangmeng Chen. Lung Nodule CT Image Classification Based on Adaptive Aggregate Weight Federated Learning[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2210003

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Nov. 11, 2022

    Accepted: Feb. 22, 2023

    Published Online: Nov. 6, 2023

    The Author Email: Feng Bao (fengbao1986.love@163.com)

    DOI:10.3788/LOP223027

    Topics