Umeå universitets logga

umu.sePublikationer
Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
A Data-Adaptive Loss Function for Incomplete Data and Incremental Learning in Semantic Image Segmentation
Umeå universitet, Medicinska fakulteten, Institutionen för strålningsvetenskaper, Radiofysik.ORCID-id: 0000-0002-2391-1419
Umeå universitet, Medicinska fakulteten, Institutionen för strålningsvetenskaper, Radiofysik.
Umeå universitet, Medicinska fakulteten, Institutionen för strålningsvetenskaper, Radiofysik.ORCID-id: 0000-0002-8971-9788
Umeå universitet, Teknisk-naturvetenskapliga fakulteten, Institutionen för datavetenskap.ORCID-id: 0000-0001-7119-7646
2022 (Engelska)Ingår i: IEEE Transactions on Medical Imaging, ISSN 0278-0062, E-ISSN 1558-254X, Vol. 41, nr 6, s. 1320-1330Artikel i tidskrift (Refereegranskat) Published
Abstract [en]

In the last years, deep learning has dramatically improved the performances in a variety of medical image analysis applications. Among different types of deep learning models, convolutional neural networks have been among the most successful and they have been used in many applications in medical imaging.

Training deep convolutional neural networks often requires large amounts of image data to generalize well to new unseen images. It is often time-consuming and expensive to collect large amounts of data in the medical image domain due to expensive imaging systems, and the need for experts to manually make ground truth annotations. A potential problem arises if new structures are added when a decision support system is already deployed and in use. Since the field of radiation therapy is constantly developing, the new structures would also have to be covered by the decision support system.

In the present work, we propose a novel loss function to solve multiple problems: imbalanced datasets, partially-labeled data, and incremental learning. The proposed loss function adapts to the available data in order to utilize all available data, even when some have missing annotations. We demonstrate that the proposed loss function also works well in an incremental learning setting, where an existing model is easily adapted to semi-automatically incorporate delineations of new organs when they appear. Experiments on a large in-house dataset show that the proposed method performs on par with baseline models, while greatly reducing the training time and eliminating the hassle of maintaining multiple models in practice.

Ort, förlag, år, upplaga, sidor
IEEE, 2022. Vol. 41, nr 6, s. 1320-1330
Nyckelord [en]
Adaptation models, Computational modeling, CT, Data models, Image segmentation, Incremental Learning, Medical Imaging, Missing Data, Predictive models, Semantic Image Segmentation, Task analysis, Training
Nationell ämneskategori
Datavetenskap (datalogi) Datorgrafik och datorseende
Identifikatorer
URN: urn:nbn:se:umu:diva-191280DOI: 10.1109/TMI.2021.3139161ISI: 000804690300003PubMedID: 34965206Scopus ID: 2-s2.0-85122295338OAI: oai:DiVA.org:umu-191280DiVA, id: diva2:1627285
Forskningsfinansiär
Vetenskapsrådet, 2018-05973Cancerforskningsfonden i Norrland, AMP 20-1027Cancerforskningsfonden i Norrland, LP 18-2182Region VästerbottenVinnovaTillgänglig från: 2022-01-13 Skapad: 2022-01-13 Senast uppdaterad: 2025-02-01Bibliografiskt granskad
Ingår i avhandling
1. Resource efficient automatic segmentation of medical images
Öppna denna publikation i ny flik eller fönster >>Resource efficient automatic segmentation of medical images
2023 (Engelska)Doktorsavhandling, sammanläggning (Övrigt vetenskapligt)
Alternativ titel[sv]
Resurseffektiv automatisk medicinsk bildsegmentering
Abstract [en]

Cancer is one of the leading causes of death worldwide. In 2020, there were around 10 million cancer deaths and nearly 20 million new cancer cases in the world. Radiation therapy is essential in cancer treatments because half of the cancer patients receive radiation therapy at some point. During a radiotherapy treatment planning (RTP), an oncologist must manually outline two types of areas of the patient’s body: target, which will be treated, and organs-at-risks (OARs), which are essential to avoid. This step is called delineation. The purpose of the delineation is to generate a sufficient dose plan that can provide adequate radiation dose to a tumor and limit the radiation exposure to healthy tissue. Therefore, accurate delineations are essential to achieve this goal.

Delineation is tedious and demanding for oncologists because it requires hours of concentrating work doing a repeated job. This is a RTP bottleneck which is often time- and resource-intensive. Current software, such as atlasbased techniques, can assist with this procedure by registering the patient’s anatomy to a predetermined anatomical map. However, the atlas-based methods are often slowed down and erroneous for patients with abnormal anatomies.

In recent years, deep learning (DL) methods, particularly convolutional neural networks (CNNs), have led to breakthroughs in numerous medical imaging applications. The core benefits of CNNs are weight sharing and that they can automatically detect important visual features. A typical application of CNNs for medical images is to automatically segment tumors, organs, and structures, which is assumed to save radiation oncologists much time when delineating. This thesis contributes to resource efficient automatic segmentation and covers different aspects of resource efficiency.

In Paper I, we proposed a novel end-to-end cascaded network for semantic segmentation in brain tumors in the multi-modal magnetic resonance imaging challenge in 2019. The proposed method used the hierarchical structure of the tumor sub-regions and was one of the top-ranking teams in the task of quantification of uncertainty in segmentation. A follow-up work to this paper was ranked second in the same task in the same challenge a year later.

We systematically assessed the segmentation performance and computational costs of the technique called pseudo-3D as a function of the number of input slices in Paper II. We compared the results to typical two-dimensional (2D) and three-dimensional (3D) CNNs and a method called triplanar orthogonal 2D. The typical pseudo-3D approach considers adjacent slices to be several image input channels. We discovered that a substantial benefit from employing multiple input slices was apparent for a specific input size.

We introduced a novel loss function in Paper III to address diverse issues, including imbalanced datasets, partially labeled data, and incremental learning. The proposed loss function adjusts to the given data to use all accessible data, even if some lack annotations. We show that the suggested loss function also performs well in an incremental learning context, where an existing model can be modified to incorporate the delineations of newly appearing organs semi-automatically.

In Paper IV, we proposed a novel method for compressing high-dimensional activation maps, which are the primary source of memory use in modern systems. We examined three distinct compression methods for the activation maps to accomplishing this. We demonstrated that the proposed method induces a regularization effect that acts on the layer weight gradients. By employing the proposed technique, we reduced activation map memory usage by up to 95%.

We investigated the use of generative adversarial networks (GANs) to enlarge a small dataset by generating synthetic images in Paper V. We use the real and generated data during training CNNs for the downstream segmentation tasks. Inspired by an existing GAN, we proposed a conditional version to generate high-dimensional and high-quality medical images of different modalities and their corresponding label maps. We evaluated the quality of the generated medical images and the effect of this augmentation on the performance of the segmentation task on six datasets.

Ort, förlag, år, upplaga, sidor
Umeå: Umeå University, 2023. s. 101
Serie
Umeå University medical dissertations, ISSN 0346-6612 ; 2219
Nyckelord
radiotherapy, medical imaging, deep learning, convolutional neural network, generative adversarial network, data augmentation, semantic segmentation, classification, activation map compression
Nationell ämneskategori
Datavetenskap (datalogi) Radiologi och bildbehandling Medicinsk bildvetenskap
Forskningsämne
radiofysik
Identifikatorer
urn:nbn:se:umu:diva-203993 (URN)978-91-7855-956-5 (ISBN)978-91-7855-957-2 (ISBN)
Disputation
2023-02-24, Bergasalen, Byggnad 27, Entré Syd, Norrlands universitetssjukhus (NUS), Umeå, 09:00 (Engelska)
Opponent
Handledare
Tillgänglig från: 2023-02-03 Skapad: 2023-01-24 Senast uppdaterad: 2025-02-09Bibliografiskt granskad

Open Access i DiVA

fulltext(2197 kB)259 nedladdningar
Filinformation
Filnamn FULLTEXT01.pdfFilstorlek 2197 kBChecksumma SHA-512
bdf6cbdc98bab363041a1de9b20a3435c2da9e4dcd3f910eec3d2a30a56884332caa5cd7befc20c29f5ba816c4b6f493f433c4ed8182769df7ba04b58ccef48c
Typ fulltextMimetyp application/pdf

Övriga länkar

Förlagets fulltextPubMedScopus

Person

Vu, Minh HoangNyholm, TufveLöfstedt, Tommy

Sök vidare i DiVA

Av författaren/redaktören
Vu, Minh HoangNyholm, TufveLöfstedt, Tommy
Av organisationen
RadiofysikInstitutionen för datavetenskap
I samma tidskrift
IEEE Transactions on Medical Imaging
Datavetenskap (datalogi)Datorgrafik och datorseende

Sök vidare utanför DiVA

GoogleGoogle Scholar
Totalt: 264 nedladdningar
Antalet nedladdningar är summan av nedladdningar för alla fulltexter. Det kan inkludera t.ex tidigare versioner som nu inte längre är tillgängliga.

doi
pubmed
urn-nbn

Altmetricpoäng

doi
pubmed
urn-nbn
Totalt: 464 träffar
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf