WAL-Net: Weakly supervised auxiliary task learning network for carotid plaques classification

Haitao Gan, Lingchao Fu, Ran Zhou, Weiyan Gan, Furong Wang +3 more
1/25/2024
eess.IVcs.CV

No official code found. Best implementation: a610lab/WAL-Net (3 stars, Unlicense, last commit 2024-06-20).

Abstract

The classification of carotid artery ultrasound images is a crucial means for diagnosing carotid plaques, holding significant clinical relevance for predicting the risk of stroke. Recent research suggests that utilizing plaque segmentation as an auxiliary task for classification can enhance performance by leveraging the correlation between segmentation and classification tasks. However, this approach relies on obtaining a substantial amount of challenging-to-acquire segmentation annotations. This paper proposes a novel weakly supervised auxiliary task learning network model (WAL-Net) to explore the interdependence between carotid plaque classification and segmentation tasks. The plaque classification task is primary task, while the plaque segmentation task serves as an auxiliary task, providing valuable information to enhance the performance of the primary task. Weakly supervised learning is adopted in the auxiliary task to completely break away from the dependence on segmentation annotations. Experiments and evaluations are conducted on a dataset comprising 1270 carotid plaque ultrasound images from Wuhan University Zhongnan Hospital. Results indicate that the proposed method achieved an approximately 1.3% improvement in carotid plaque classification accuracy compared to the baseline network. Specifically, the accuracy of mixed-echoic plaques classification increased by approximately 3.3%, demonstrating the effectiveness of our approach.

View on arXivView PDF

Code Implementations(1)

a610lab/WAL-NetCommunity100%
Unlicense
31PythonJan 27, 20242 years agoUnlicense

Similar Papers

Cite this paper

@article{gan2024walnet,
  title  = {WAL-Net: Weakly supervised auxiliary task learning network for carotid plaques classification},
  author = {Haitao Gan and Lingchao Fu and Ran Zhou and Weiyan Gan and Furong Wang and Xiaoyan Wu and Zhi Yang and Zhongwei Huang},
  year   = {2024},
  eprint = {2401.13998},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.13998v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.13998.svg)](https://codeofpaper.com/paper/2401.13998)

Discussion