kth.sePublications KTH
Change search
Link to record
Permanent link

Direct link
Publications (10 of 12) Show all publications
Sorkhei, M., Konuk, E., Guo, J., Meng, C., Matsoukas, C. & Smith, K. (2025). Efficient Self-Supervised Adaptation for Medical Image Analysis. In: Proceedings 2025 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW): . Paper presented at nternational Conference on Computer Vision, ICCV 2025 CVAMD, Honolulu, Hawai'i, October 19-23, 2025. Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Efficient Self-Supervised Adaptation for Medical Image Analysis
Show others...
2025 (English)In: Proceedings 2025 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW), Institute of Electrical and Electronics Engineers (IEEE) , 2025Conference paper, Published paper (Refereed)
Abstract [en]

Self-supervised adaptation (SSA) improves foundation model transfer to medical domains but is computationally prohibitive. Although parameter efficient fine-tuning (PEFT) methods such as LoRA have been explored for supervised adaptation, their effectiveness for SSA remains unknown. In this work, we introduce efficient self-supervised adaptation (ESSA), a framework that applies parameter-efficient fine-tuning techniques to SSA with the aim of reducing computational cost and improving adaptation performance. To the best of our knowledge, we are the first to demonstrate that PEFT methods can be effectively applied to SSA to improve self-supervised learning, challenging the assumption that full-parameter SSA is necessary for optimal performance. Furthermore, we show that applying PEFT during supervised adaptation following self-supervision leads to additional performance gains, outperforming full-parameter training.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2025
National Category
Computer graphics and computer vision
Identifiers
urn:nbn:se:kth:diva-380658 (URN)10.1109/ICCVW69036.2025.00096 (DOI)001740020100093 ()2-s2.0-105035176775 (Scopus ID)
Conference
nternational Conference on Computer Vision, ICCV 2025 CVAMD, Honolulu, Hawai'i, October 19-23, 2025
Note

Part of ISBN 979-8-3315-8988-2

QC 20260717

Available from: 2026-05-04 Created: 2026-05-04 Last updated: 2026-07-17Bibliographically approved
Sorkhei, M., Matsoukas, C., Fredin Haslum, J., Konuk, E. & Smith, K. (2025). k-NN as a Simple and Effective Estimator of Transferability. Transactions on Machine Learning Research, 2025-October
Open this publication in new window or tab >>k-NN as a Simple and Effective Estimator of Transferability
Show others...
2025 (English)In: Transactions on Machine Learning Research, E-ISSN 2835-8856, Vol. 2025-OctoberArticle in journal (Refereed) Published
Abstract [en]

How well can one expect transfer learning to work in a new setting where the domain is shifted, the task is different, and the architecture changes? Many transfer learning metrics have been proposed to answer this question. But how accurate are their predictions in a realistic new setting? We conducted an extensive evaluation involving over 42,000 experiments comparing 23 transferability metrics across 16 different datasets to assess their ability to predict transfer performance for image classification tasks. Our findings reveal that none of the existing metrics perform well across the board. However, we find that a simple k-nearest neighbor evaluation – as is commonly used to evaluate feature quality for self-supervision – not only surpasses existing metrics, but also offers better computational efficiency and ease of implementation.

Place, publisher, year, edition, pages
Transactions on Machine Learning Research, 2025
National Category
Computer graphics and computer vision Computer Sciences
Identifiers
urn:nbn:se:kth:diva-372408 (URN)2-s2.0-105018634464 (Scopus ID)
Note

QC 20251106

Available from: 2025-11-06 Created: 2025-11-06 Last updated: 2026-05-07Bibliographically approved
Salim, M., Liu, Y., Sorkhei, M., Ntoula, D., Foukakis, T., Fredriksson, I., . . . Strand, F. (2024). AI-based selection of individuals for supplemental MRI in population-based breast cancer screening: the randomized ScreenTrustMRI trial. Nature Medicine, 30(9), 2623-2630
Open this publication in new window or tab >>AI-based selection of individuals for supplemental MRI in population-based breast cancer screening: the randomized ScreenTrustMRI trial
Show others...
2024 (English)In: Nature Medicine, ISSN 1078-8956, E-ISSN 1546-170X, Vol. 30, no 9, p. 2623-2630Article in journal (Refereed) Published
Abstract [en]

Screening mammography reduces breast cancer mortality, but studies analyzing interval cancers diagnosed after negative screens have shown that many cancers are missed. Supplemental screening using magnetic resonance imaging (MRI) can reduce the number of missed cancers. However, as qualified MRI staff are lacking, the equipment is expensive to purchase and cost-effectiveness for screening may not be convincing, the utilization of MRI is currently limited. An effective method for triaging individuals to supplemental MRI screening is therefore needed. We conducted a randomized clinical trial, ScreenTrustMRI, using a recently developed artificial intelligence (AI) tool to score each mammogram. We offered trial participation to individuals with a negative screening mammogram and a high AI score (top 6.9%). Upon agreeing to participate, individuals were assigned randomly to one of two groups: those receiving supplemental MRI and those not receiving MRI. The primary endpoint of ScreenTrustMRI is advanced breast cancer defined as either interval cancer, invasive component larger than 15 mm or lymph node positive cancer, based on a 27-month follow-up time from the initial screening. Secondary endpoints, prespecified in the study protocol to be reported before the primary outcome, include cancer detected by supplemental MRI, which is the focus of the current paper. Compared with traditional breast density measures used in a previous clinical trial, the current AI method was nearly four times more efficient in terms of cancers detected per 1,000 MRI examinations (64 versus 16.5). Most additional cancers detected were invasive and several were multifocal, suggesting that their detection was timely. Altogether, our results show that using an AI-based score to select a small proportion (6.9%) of individuals for supplemental MRI after negative mammography detects many missed cancers, making the cost per cancer detected comparable with screening mammography. ClinicalTrials.gov registration: NCT04832594.

Place, publisher, year, edition, pages
Nature Research, 2024
National Category
Radiology and Medical Imaging Cancer and Oncology Computational Mathematics
Identifiers
urn:nbn:se:kth:diva-366605 (URN)10.1038/s41591-024-03093-5 (DOI)001264739500004 ()38977914 (PubMedID)2-s2.0-85197684683 (Scopus ID)
Note

QC 20250709

Available from: 2025-07-09 Created: 2025-07-09 Last updated: 2025-07-09Bibliographically approved
Konuk, E., Matsoukas, C., Sorkhei, M., Lertsiravarameth, P. & Smith, K. (2024). Learning from Offline Foundation Features with Tensor Augmentations. In: A. Globerson, L. Mackey, D. Belgrave, A. Fan, U. Paquet, J. Tomczak and C. Zhang (Ed.), Advances in Neural Information Processing Systems 37 (NeurIPS 2024): . Paper presented at NeurIPS 2024, the Thirty-Eighth Annual Conference on Neural Information Processing Systems, Vancouver, December 10-15, 2024. Curran Associates
Open this publication in new window or tab >>Learning from Offline Foundation Features with Tensor Augmentations
Show others...
2024 (English)In: Advances in Neural Information Processing Systems 37 (NeurIPS 2024) / [ed] A. Globerson, L. Mackey, D. Belgrave, A. Fan, U. Paquet, J. Tomczak and C. Zhang, Curran Associates , 2024Conference paper, Published paper (Refereed)
Abstract [en]

We introduce Learning from Offline Foundation Features with Tensor Augmentations (LOFF-TA), an efficient training scheme designed to harness the capabilities of foundation models in limited resource settings where their direct development is not feasible. LOFF-TA involves training a compact classifier on cached feature embeddings from a frozen foundation model, resulting in up to 37× faster training and up to 26× reduced GPU memory usage. Because the embeddings of augmented images would be too numerous to store, yet the augmentation process is essential for training, we propose to apply tensor augmentations to the cached embeddings of the original non-augmented images. LOFF-TA makes it possible to leverage the power of foundation models, regardless of their size, in settings with limited computational capacity. Moreover, LOFF-TA can be used to apply foundation models to high-resolution images without increasing compute. In certain scenarios, we find that training with LOFF-TA yields better results than directly fine-tuning the foundation model.

Place, publisher, year, edition, pages
Curran Associates, 2024
Keywords
Adaptation, Transfer learning, Foundation models, Augmentation
National Category
Computer graphics and computer vision
Research subject
Computer Science
Identifiers
urn:nbn:se:kth:diva-354832 (URN)2-s2.0-105000782383 (Scopus ID)
Conference
NeurIPS 2024, the Thirty-Eighth Annual Conference on Neural Information Processing Systems, Vancouver, December 10-15, 2024
Funder
Wallenberg AI, Autonomous Systems and Software Program (WASP)
Note

QC 20250408

Available from: 2024-10-14 Created: 2024-10-14 Last updated: 2025-04-08Bibliographically approved
Liu, Y., Sorkhei, M., Dembrower, K., Azizpour, H., Strand, F. & Smith, K. (2024). Use of an AI Score Combining Cancer Signs, Masking, and Risk to Select Patients for Supplemental Breast Cancer Screening. Radiology, 311(1), Article ID e232535.
Open this publication in new window or tab >>Use of an AI Score Combining Cancer Signs, Masking, and Risk to Select Patients for Supplemental Breast Cancer Screening
Show others...
2024 (English)In: Radiology, ISSN 0033-8419, E-ISSN 1527-1315, Vol. 311, no 1, article id e232535Article in journal (Refereed) Published
Abstract [en]

Background: Mammographic density measurements are used to identify patients who should undergo supplemental imaging for breast cancer detection, but artificial intelligence (AI) image analysis may be more effective.<br /> Purpose: To assess whether AISmartDensity-an AI -based score integrating cancer signs, masking, and risk-surpasses measurements of mammographic density in identifying patients for supplemental breast imaging after a negative screening mammogram. Materials and Methods: This retrospective study included randomly selected individuals who underwent screening mammography at Karolinska University Hospital between January 2008 and December 2015. The models in AISmartDensity were trained and validated using nonoverlapping data. The ability of AISmartDensity to identify future cancer in patients with a negative screening mammogram was evaluated and compared with that of mammographic density models. Sensitivity and positive predictive value (PPV) were calculated for the top 8% of scores, mimicking the proportion of patients in the Breast Imaging Reporting and Data System "extremely dense" category. Model performance was evaluated using area under the receiver operating characteristic curve (AUC) and was compared using the DeLong test.<br /> Results: The study population included 65 325 examinations (median patient age, 53 years [IQR, 47-62 years])-64 870 examinations in healthy patients and 455 examinations in patients with breast cancer diagnosed within 3 years of a negative screening mammogram. The AUC for detecting subsequent cancers was 0.72 and 0.61 ( P < .001) for AISmartDensity and the best -performing density model (age -adjusted dense area), respectively. For examinations with scores in the top 8%, AISmartDensity identified 152 of 455 (33%) future cancers with a PPV of 2.91%, whereas the best -performing density model (age -adjusted dense area) identified 57 of 455 (13%) future cancers with a PPV of 1.09% ( P < .001). AISmartDensity identified 32% (41 of 130) and 34% (111 of 325) of interval and next -round screen -detected cancers, whereas the best -performing density model (dense area) identified 16% (21 of 130) and 9% (30 of 325), respectively.<br /> Conclusion: AISmartDensity, integrating cancer signs, masking, and risk, outperformed traditional density models in identifying patients for supplemental imaging after a negative screening mammogram.

Place, publisher, year, edition, pages
Radiological Society of North America (RSNA), 2024
National Category
Cancer and Oncology
Identifiers
urn:nbn:se:kth:diva-349627 (URN)10.1148/radiol.232535 (DOI)001245823000007 ()38591971 (PubMedID)2-s2.0-85190324943 (Scopus ID)
Note

QC 20240702

Available from: 2024-07-02 Created: 2024-07-02 Last updated: 2024-07-02Bibliographically approved
Liu, Y., Sorkhei, M., Dembrower, K., Azizpour, H., Strand, F. & Smith, K. (2023). Selecting Women for Supplemental Breast Imaging using AI Biomarkers of Cancer Signs, Masking, and Risk.
Open this publication in new window or tab >>Selecting Women for Supplemental Breast Imaging using AI Biomarkers of Cancer Signs, Masking, and Risk
Show others...
2023 (English)Manuscript (preprint) (Other academic)
Abstract [en]

Background: Traditional mammographic density aids in determining the need for supplemental imagingby MRI or ultrasound. However, AI image analysis, considering more subtle and complex image features,may enable a more effective identification of women requiring supplemental imaging.Purpose: To assess if AISmartDensity, an AI-based score considering cancer signs, masking, and risk,surpasses traditional mammographic density in identifying women for supplemental imaging after negativescreening mammography.Methods: This retrospective study included randomly selected breast cancer patients and healthy controlsat Karolinska University Hospital between 2008 and 2015. Bootstrapping simulated a 0.2% interval cancerrate. We included previous exams for diagnosed women and all exams for controls. AISmartDensity hadbeen developed using random mammograms from a population non-overlapping with the current studypopulation. We evaluated AISmartDensity to, based on negative screening mammograms, identify womenwith interval cancer and next-round screen-detected cancer. It was compared to age and density models, withsensitivity and PPV calculated for women with the top 8% scores, mimicking the proportion of BIRADS“extremely dense” category. Statistical significance was determined using the Student’s t-test.Results: The study involved 2043 women, 258 with breast cancer diagnosed within 3 years of a negativemammogram, and 1785 healthy controls. Diagnosed women had a median age of 57 years (IQR 16) versus53 years (IQR 15) for controls (p < .001). At the 92nd percentile, AISmartDenstiy identified 87 (33.67%)future cancers with PPV 1.68%, whereas mammographic density identified 34 (13.18%) with PPV 0.66%(p < .001). AISmartDensity identified 32% interval and 36% next-round cancers, versus mammographicdensity’s 16% and 10%. The combined mammographic density and age model yielded an AUC of 0.60,significantly lower than AISmartDensity’s 0.73 (p < .001).Conclusions: AISmartDensity, integrating cancer signs, masking, and risk, more effectively identifiedwomen for additional breast imaging than traditional age and density models. 

National Category
Medical and Health Sciences Engineering and Technology
Identifiers
urn:nbn:se:kth:diva-340721 (URN)
Note

QC 20231218

Available from: 2023-12-11 Created: 2023-12-11 Last updated: 2023-12-18Bibliographically approved
Matsoukas, C., Fredin Haslum, J., Sorkhei, M., Soderberg, M. & Smith, K. (2022). What Makes Transfer Learning Work for Medical Images: Feature Reuse & Other Factors. In: 2022 IEEE/CVF conference on computer vision and pattern recognition (CVPR): . Paper presented at IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), JUN 18-24, 2022, New Orleans, LA (pp. 9215-9224). Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>What Makes Transfer Learning Work for Medical Images: Feature Reuse & Other Factors
Show others...
2022 (English)In: 2022 IEEE/CVF conference on computer vision and pattern recognition (CVPR), Institute of Electrical and Electronics Engineers (IEEE) , 2022, p. 9215-9224Conference paper, Published paper (Refereed)
Abstract [en]

Transfer learning is a standard technique to transfer knowledge from one domain to another. For applications in medical imaging, transfer from ImageNet has become the de-facto approach, despite differences in the tasks and image characteristics between the domains. However, it is unclear what factors determine whether - and to what extent transfer learning to the medical domain is useful. The longstanding assumption that features from the source domain get reused has recently been called into question. Through a series of experiments on several medical image benchmark datasets, we explore the relationship between transfer learning, data size, the capacity and inductive bias of the model, as well as the distance between the source and target domain. Our findings suggest that transfer learning is beneficial in most cases, and we characterize the important role feature reuse plays in its success.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2022
Series
IEEE Conference on Computer Vision and Pattern Recognition, ISSN 1063-6919
National Category
Computational Mathematics
Identifiers
urn:nbn:se:kth:diva-322794 (URN)10.1109/CVPR52688.2022.00901 (DOI)000870759102028 ()2-s2.0-85137378486 (Scopus ID)
Conference
IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), JUN 18-24, 2022, New Orleans, LA
Note

Part of proceedings ISBN 978-1-6654-6946-3

QC 20230131

Available from: 2023-01-31 Created: 2023-01-31 Last updated: 2024-05-20Bibliographically approved
Sorkhei, M., Liu, Y., Azizpour, H., Azavedo, E., Dembrower, K., Ntoula, D., . . . Smith, K. (2021). CSAW-M: An Ordinal Classification Dataset for Benchmarking Mammographic Masking of Cancer. In: Proceedings of the Neural Information Processing Systems Track on Datasets and Benchmarks 1, NeurIPS Datasets and Benchmarks 2021: . Paper presented at 35th Conference on Neural Information Processing Systems - Track on Datasets and Benchmarks, NeurIPS Datasets and Benchmarks 2021, Virtual, Online, NA, Dec 6 2021 - Dec 14 2021. Neural Information Processing Systems Foundation
Open this publication in new window or tab >>CSAW-M: An Ordinal Classification Dataset for Benchmarking Mammographic Masking of Cancer
Show others...
2021 (English)In: Proceedings of the Neural Information Processing Systems Track on Datasets and Benchmarks 1, NeurIPS Datasets and Benchmarks 2021, Neural Information Processing Systems Foundation , 2021Conference paper, Published paper (Refereed)
Abstract [en]

Interval and large invasive breast cancers, which are associated with worse prognosis than other cancers, are usually detected at a late stage due to false negative assessments of screening mammograms. The missed screening-time detection is commonly caused by the tumor being obscured by its surrounding breast tissues, a phenomenon called masking. To study and benchmark mammographic masking of cancer, in this work we introduce CSAW-M, the largest public mammographic dataset, collected from over 10,000 individuals and annotated with potential masking. In contrast to the previous approaches which measure breast image density as a proxy, our dataset directly provides annotations of masking potential assessments from five specialists. We also trained deep learning models on CSAW-M to estimate the masking level and showed that the estimated masking is significantly more predictive of screening participants diagnosed with interval and large invasive cancers – without being explicitly trained for these tasks – than its breast density counterparts.

Place, publisher, year, edition, pages
Neural Information Processing Systems Foundation, 2021
National Category
Cancer and Oncology Radiology and Medical Imaging
Identifiers
urn:nbn:se:kth:diva-361967 (URN)2-s2.0-105000231004 (Scopus ID)
Conference
35th Conference on Neural Information Processing Systems - Track on Datasets and Benchmarks, NeurIPS Datasets and Benchmarks 2021, Virtual, Online, NA, Dec 6 2021 - Dec 14 2021
Note

QC 20250404

Available from: 2025-04-03 Created: 2025-04-03 Last updated: 2025-04-04Bibliographically approved
Sorkhei, M., Liu, Y., Azizpour, H., Azavedo, E., Dembrower, K., Ntoula, D., . . . Smith, K. (2021). CSAW-M: An Ordinal Classification Dataset for Benchmarking Mammographic Masking of Cancer. In: Conference on Neural Information Processing Systems (NeurIPS) – Datasets and Benchmarks Proceedings, 2021.: . Paper presented at 35th Conference on Neural Information Processing Systems (NeurIPS 2021) Track on Datasets and Benchmarks, 6-14 Dec 2021, virtual..
Open this publication in new window or tab >>CSAW-M: An Ordinal Classification Dataset for Benchmarking Mammographic Masking of Cancer
Show others...
2021 (English)In: Conference on Neural Information Processing Systems (NeurIPS) – Datasets and Benchmarks Proceedings, 2021., 2021Conference paper, Published paper (Refereed)
Abstract [en]

Interval and large invasive breast cancers, which are associated with worse prognosis than other cancers, are usually detected at a late stage due to false negative assessments of screening mammograms. The missed screening-time detection is commonly caused by the tumor being obscured by its surrounding breast tissues, a phenomenon called masking. To study and benchmark mammographic masking of cancer, in this work we introduce CSAW-M, the largest public mammographic dataset, collected from over 10,000 individuals and annotated with potential masking. In contrast to the previous approaches which measure breast image density as a proxy, our dataset directly provides annotations of masking potential assessments from five specialists. We also trained deep learning models on CSAW-M to estimate the masking level and showed that the estimated masking is significantly more predictive of screening participants diagnosed with interval and large invasive cancers – without being explicitly trained for these tasks – than its breast density counterparts.

National Category
Engineering and Technology
Identifiers
urn:nbn:se:kth:diva-340718 (URN)
Conference
35th Conference on Neural Information Processing Systems (NeurIPS 2021) Track on Datasets and Benchmarks, 6-14 Dec 2021, virtual.
Note

QC 20231218

Available from: 2023-12-11 Created: 2023-12-11 Last updated: 2026-05-07Bibliographically approved
Sorkhei, M. M., Henter, G. E. & Kjellström, H. (2021). Full-Glow: Fully conditional Glow for more realistic image generation. In: Bauckhage, C., Gall, J., Schwing, A. (Ed.), Pattern Recognition: 43rd DAGM German Conference, DAGM GCPR 2021. Paper presented at 43rd DAGM German Conference on Pattern Recognition, DAGM GCPR 2021, Virtual, Online, 28 September 2021 through 1 October 2021 (pp. 697-711). Cham, Switzerland: Springer Nature, 13024
Open this publication in new window or tab >>Full-Glow: Fully conditional Glow for more realistic image generation
2021 (English)In: Pattern Recognition: 43rd DAGM German Conference, DAGM GCPR 2021 / [ed] Bauckhage, C., Gall, J., Schwing, A., Cham, Switzerland: Springer Nature , 2021, Vol. 13024, p. 697-711Conference paper, Published paper (Refereed)
Abstract [en]

Autonomous agents, such as driverless cars, require large amounts of labeled visual data for their training. A viable approach for acquiring such data is training a generative model with collected real data, and then augmenting the collected real dataset with synthetic images from the model, generated with control of the scene layout and ground truth labeling. In this paper we propose Full-Glow, a fully conditional Glow-based architecture for generating plausible and realistic images of novel street scenes given a semantic segmentation map indicating the scene layout. Benchmark comparisons show our model to outperform recent works in terms of the semantic segmentation performance of a pretrained PSPNet. This indicates that images from our model are, to a higher degree than from other models, similar to real images of the same kinds of scenes and objects, making them suitable as training data for a visual semantic segmentation or object recognition system.

Place, publisher, year, edition, pages
Cham, Switzerland: Springer Nature, 2021
Series
Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), ISSN 0302-9743 ; 13024
Keywords
Conditional image generation, generative models, normalizing flows
National Category
Computer and Information Sciences
Research subject
Computer Science
Identifiers
urn:nbn:se:kth:diva-312897 (URN)10.1007/978-3-030-92659-5_45 (DOI)001500565200045 ()2-s2.0-85124290582 (Scopus ID)
Conference
43rd DAGM German Conference on Pattern Recognition, DAGM GCPR 2021, Virtual, Online, 28 September 2021 through 1 October 2021
Funder
Wallenberg AI, Autonomous Systems and Software Program (WASP)
Note

QC 20220530

Part of proceedings: ISBN 978-303092658-8

Available from: 2022-05-24 Created: 2022-05-24 Last updated: 2025-12-05Bibliographically approved
Organisations
Identifiers
ORCID iD: ORCID iD iconorcid.org/0000-0001-6204-0778

Search in DiVA

Show all publications