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

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.

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


  • 309


  • 1


Share Link