Contrastive-Adversarial and Diffusion: Exploring pre-training and fine-tuning strategies for sulcal identification

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Mamalakis, Michail, de Vareilles, Héloïse, Wu, Shun-Chin Jim, Agartz, Ingrid, Mørch-Johnsen, Lynn Egeland, Garrison, Jane, Simons, Jon, Lio, Pietro, Suckling, John, Murray, Graham
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929364365475840
author Mamalakis, Michail
de Vareilles, Héloïse
Wu, Shun-Chin Jim
Agartz, Ingrid
Mørch-Johnsen, Lynn Egeland
Garrison, Jane
Simons, Jon
Lio, Pietro
Suckling, John
Murray, Graham
author_facet Mamalakis, Michail
de Vareilles, Héloïse
Wu, Shun-Chin Jim
Agartz, Ingrid
Mørch-Johnsen, Lynn Egeland
Garrison, Jane
Simons, Jon
Lio, Pietro
Suckling, John
Murray, Graham
contents In the last decade, computer vision has witnessed the establishment of various training and learning approaches. Techniques like adversarial learning, contrastive learning, diffusion denoising learning, and ordinary reconstruction learning have become standard, representing state-of-the-art methods extensively employed for fully training or pre-training networks across various vision tasks. The exploration of fine-tuning approaches has emerged as a current focal point, addressing the need for efficient model tuning with reduced GPU memory usage and time costs while enhancing overall performance, as exemplified by methodologies like low-rank adaptation (LoRA). Key questions arise: which pre-training technique yields optimal results - adversarial, contrastive, reconstruction, or diffusion denoising? How does the performance of these approaches vary as the complexity of fine-tuning is adjusted? This study aims to elucidate the advantages of pre-training techniques and fine-tuning strategies to enhance the learning process of neural networks in independent identical distribution (IID) cohorts. We underscore the significance of fine-tuning by examining various cases, including full tuning, decoder tuning, top-level tuning, and fine-tuning of linear parameters using LoRA. Systematic summaries of model performance and efficiency are presented, leveraging metrics such as accuracy, time cost, and memory efficiency. To empirically demonstrate our findings, we focus on a multi-task segmentation-classification challenge involving the paracingulate sulcus (PCS) using different 3D Convolutional Neural Network (CNN) architectures by using the TOP-OSLO cohort comprising 596 subjects.
format Preprint
id arxiv_https___arxiv_org_abs_2405_19204
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Contrastive-Adversarial and Diffusion: Exploring pre-training and fine-tuning strategies for sulcal identification
Mamalakis, Michail
de Vareilles, Héloïse
Wu, Shun-Chin Jim
Agartz, Ingrid
Mørch-Johnsen, Lynn Egeland
Garrison, Jane
Simons, Jon
Lio, Pietro
Suckling, John
Murray, Graham
Image and Video Processing
Computer Vision and Pattern Recognition
In the last decade, computer vision has witnessed the establishment of various training and learning approaches. Techniques like adversarial learning, contrastive learning, diffusion denoising learning, and ordinary reconstruction learning have become standard, representing state-of-the-art methods extensively employed for fully training or pre-training networks across various vision tasks. The exploration of fine-tuning approaches has emerged as a current focal point, addressing the need for efficient model tuning with reduced GPU memory usage and time costs while enhancing overall performance, as exemplified by methodologies like low-rank adaptation (LoRA). Key questions arise: which pre-training technique yields optimal results - adversarial, contrastive, reconstruction, or diffusion denoising? How does the performance of these approaches vary as the complexity of fine-tuning is adjusted? This study aims to elucidate the advantages of pre-training techniques and fine-tuning strategies to enhance the learning process of neural networks in independent identical distribution (IID) cohorts. We underscore the significance of fine-tuning by examining various cases, including full tuning, decoder tuning, top-level tuning, and fine-tuning of linear parameters using LoRA. Systematic summaries of model performance and efficiency are presented, leveraging metrics such as accuracy, time cost, and memory efficiency. To empirically demonstrate our findings, we focus on a multi-task segmentation-classification challenge involving the paracingulate sulcus (PCS) using different 3D Convolutional Neural Network (CNN) architectures by using the TOP-OSLO cohort comprising 596 subjects.
title Contrastive-Adversarial and Diffusion: Exploring pre-training and fine-tuning strategies for sulcal identification
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.19204