Open Access iconOpen Access



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


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.


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.


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.
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
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.

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


  • 511


  • 1


Share Link