A Lightweight Neural Architecture Search Model for Medical Image Classification
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866916236812615680 |
|---|---|
| author | Xie, Lunchen Lomurno, Eugenio Gambella, Matteo Ardagna, Danilo Roveri, Manuel Matteucci, Matteo Shi, Qingjiang |
| author_facet | Xie, Lunchen Lomurno, Eugenio Gambella, Matteo Ardagna, Danilo Roveri, Manuel Matteucci, Matteo Shi, Qingjiang |
| contents | Accurate classification of medical images is essential for modern diagnostics. Deep learning advancements led clinicians to increasingly use sophisticated models to make faster and more accurate decisions, sometimes replacing human judgment. However, model development is costly and repetitive. Neural Architecture Search (NAS) provides solutions by automating the design of deep learning architectures. This paper presents ZO-DARTS+, a differentiable NAS algorithm that improves search efficiency through a novel method of generating sparse probabilities by bi-level optimization. Experiments on five public medical datasets show that ZO-DARTS+ matches the accuracy of state-of-the-art solutions while reducing search times by up to three times. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_03462 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | A Lightweight Neural Architecture Search Model for Medical Image Classification Xie, Lunchen Lomurno, Eugenio Gambella, Matteo Ardagna, Danilo Roveri, Manuel Matteucci, Matteo Shi, Qingjiang Computer Vision and Pattern Recognition Artificial Intelligence Machine Learning Accurate classification of medical images is essential for modern diagnostics. Deep learning advancements led clinicians to increasingly use sophisticated models to make faster and more accurate decisions, sometimes replacing human judgment. However, model development is costly and repetitive. Neural Architecture Search (NAS) provides solutions by automating the design of deep learning architectures. This paper presents ZO-DARTS+, a differentiable NAS algorithm that improves search efficiency through a novel method of generating sparse probabilities by bi-level optimization. Experiments on five public medical datasets show that ZO-DARTS+ matches the accuracy of state-of-the-art solutions while reducing search times by up to three times. |
| title | A Lightweight Neural Architecture Search Model for Medical Image Classification |
| topic | Computer Vision and Pattern Recognition Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2405.03462 |