Open Access iconOpen Access

ARTICLE

crossmark

SW-Net: A novel few-shot learning approach for disease subtype prediction

YUHAN JI1, YONG LIANG1,*, ZIYI YANG2, NING AI1

1 Faculty of Innovation Engineering, School of Computer Science and Engineering, Macau University of Science and Technology, Macau, 999078, China
2 Tencent Quantum Lab, Shenzhen, 518000, China

* Corresponding Author: YONG LIANG. Email: email

(This article belongs to the Special Issue: Application of Deep Learning in Cancer)

BIOCELL 2023, 47(3), 569-579. https://doi.org/10.32604/biocell.2023.025865

Abstract

Few-shot learning is becoming more and more popular in many fields, especially in the computer vision field. This inspires us to introduce few-shot learning to the genomic field, which faces a typical few-shot problem because some tasks only have a limited number of samples with high-dimensions. The goal of this study was to investigate the few-shot disease sub-type prediction problem and identify patient subgroups through training on small data. Accurate disease sub-type classification allows clinicians to efficiently deliver investigations and interventions in clinical practice. We propose the SW-Net, which simulates the clinical process of extracting the shared knowledge from a range of interrelated tasks and generalizes it to unseen data. Our model is built upon a simple baseline, and we modified it for genomic data. Support-based initialization for the classifier and transductive fine-tuning techniques were applied in our model to improve prediction accuracy, and an Entropy regularization term on the query set was appended to reduce over-fitting. Moreover, to address the high dimension and high noise issue, we future extended a feature selection module to adaptively select important features and a sample weighting module to prioritize high-confidence samples. Experiments on simulated data and The Cancer Genome Atlas meta-dataset show that our new baseline model gets higher prediction accuracy compared to other competing algorithms.

Keywords


Cite This Article

APA Style
JI, Y., LIANG, Y., YANG, Z., AI, N. (2023). Sw-net: A novel few-shot learning approach for disease subtype prediction. BIOCELL, 47(3), 569-579. https://doi.org/10.32604/biocell.2023.025865
Vancouver Style
JI Y, LIANG Y, YANG Z, AI N. Sw-net: A novel few-shot learning approach for disease subtype prediction. BIOCELL . 2023;47(3):569-579 https://doi.org/10.32604/biocell.2023.025865
IEEE Style
Y. JI, Y. LIANG, Z. YANG, and N. AI "SW-Net: A novel few-shot learning approach for disease subtype prediction," BIOCELL , vol. 47, no. 3, pp. 569-579. 2023. https://doi.org/10.32604/biocell.2023.025865



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1131

    View

  • 511

    Download

  • 1

    Like

Share Link