| Literature DB >> 36266626 |
Chengkai Piao1, Mengyue Lv2, Shujie Wang2, Rongyan Zhou2, Yuchen Wang1, Jinmao Wei3, Jian Liu4.
Abstract
Recently, Deep Learning based automatic generation of treatment recommendation has been attracting much attention. However, medical datasets are usually small, which may lead to over-fitting and inferior performances of deep learning models. In this paper, we propose multi-objective data enhancement method to indirectly scale up the medical data to avoid over-fitting and generate high quantity treatment recommendations. Specifically, we define a main and several auxiliary tasks on the same dataset and train a specific model for each of these tasks to learn different aspects of knowledge in limited data scale. Meanwhile, a Soft Parameter Sharing method is exploited to share learned knowledge among models. By sharing the knowledge learned by auxiliary tasks to the main task, the proposed method can take different semantic distributions into account during the training process of the main task. We collected an ultrasound dataset of thyroid nodules that contains Findings, Impressions and Treatment Recommendations labeled by professional doctors. We conducted various experiments on the dataset to validate the proposed method and justified its better performance than existing methods.Entities:
Keywords: Deep learning; Multi-objective; Parameter sharing; Thyroid nodules; Ultrasound analysis
Mesh:
Year: 2022 PMID: 36266626 PMCID: PMC9583467 DOI: 10.1186/s12859-022-04985-4
Source DB: PubMed Journal: BMC Bioinformatics ISSN: 1471-2105 Impact factor: 3.307