1 |
[pt] AJUSTE FINO DE MODELO AUTO-SUPERVISIONADO USANDO REDES NEURAIS SIAMESAS PARA CLASSIFICAÇÃO DE IMAGENS DE COVID-19 / [en] FINE-TUNING SELF-SUPERVISED MODEL WITH SIAMESE NEURAL NETWORKS FOR COVID-19 IMAGE CLASSIFICATIONANTONIO MOREIRA PINTO 03 December 2024 (has links)
[pt] Nos últimos anos, o aprendizado auto-supervisionado demonstrou desempenho estado da arte em áreas como visão computacional e processamento de
linguagem natural. No entanto, ajustar esses modelos para tarefas específicas
de classificação, especialmente com dados rotulados, permanece sendo um desafio. Esta dissertação apresenta uma abordagem para ajuste fino de modelos
auto-supervisionados usando Redes Neurais Siamesas, aproveitando a função
de perda semi-hard triplet loss. Nosso método visa refinar as representações
do espaço latente dos modelos auto-supervisionados para melhorar seu desempenho em tarefas posteriores de classificação. O framework proposto emprega
Masked Autoencoders para pré-treinamento em um conjunto abrangente de
dados de radiografias, seguido de ajuste fino com redes siamesas para separação eficaz de características e melhor classificação. A abordagem é avaliada
no conjunto de dados COVIDx 9 para detecção de COVID-19 a partir de radiografias frontais de peito, alcançando uma nova precisão recorde de 98,5 por cento,
superando as técnicas tradicionais de ajuste fino e o modelo COVID-Net CRX
3. Os resultados demonstram a eficácia de nosso método em aumentar a utilidade de modelos auto-supervisionados para tarefas complexas de imagem
médica. Trabalhos futuros explorarão a escalabilidade dessa abordagem para
outros domínios e a integração de funções de perda de espaço de embedding
mais sofisticadas. / [en] In recent years, self-supervised learning has demonstrated state-of-theart performance in domains such as computer vision and natural language processing. However, fine-tuning these models for specific classification tasks,
particularly with labeled data, remains challenging. This thesis introduces a
novel approach to fine-tuning self-supervised models using Siamese Neural
Networks, specifically leveraging a semi-hard triplet loss function. Our method
aims to refine the latent space representations of self-supervised models to
improve their performance on downstream classification tasks. The proposed
framework employs Masked Autoencoders for pre-training on a comprehensive
radiograph dataset, followed by fine-tuning with Siamese networks for effective
feature separation and improved classification. The approach is evaluated on
the COVIDx dataset for COVID-19 detection from frontal chest radiographs,
achieving a new record accuracy of 98.5 percent, surpassing traditional fine-tuning
techniques and COVID-Net CRX 3. The results demonstrate the effectiveness
of our method in enhancing the utility of self-supervised models for complex
medical imaging tasks. Future work will explore the scalability of this approach
to other domains and the integration of more sophisticated embedding-space
loss functions.
|
Page generated in 0.0469 seconds