Cette recherche a été réalisée conformément aux directives institutionnelles pour la recherche computationnelle et le traitement des données. Aucun sujet humain ni vertébré n’a participé à cette étude.
Préparation et prétraitement des jeux de données
Collecte et organisation des jeux de données
Des images dermoscopiques ont été collectées à partir des ensembles de données PH2, ISIC2018 et ISIC2017, avec des liens spécifiques fournis dans le Table of Materials. Le jeu de données PH2 contient 200 images haute résolution (1000 × 1000 pixels), ISIC2018 comprend 2 594 images avec des masques de segmentation correspondants, et ISIC2017 comprend 2 150 images avec des masques de segmentation. Les données étaient organisées en ensembles d’entraînement, de validation et de test selon des protocoles standards, garantissant que les images sont dans des formats compatibles (par exemple, .jpg, .png ou .tiff) et incluent des masques de vérité au sol correspondants au format binaire.
Les données étaient organisées en ensembles d’entraînement, de validation et de test selon des protocoles standards. Les ensembles de données étaient répartis comme suit : pour ISIC2018, 1 815 images ont été allouées pour l’entraînement, 259 pour la validation et 520 pour les tests. Pour ISIC2017, 1 500 images ont été allouées à l’entraînement, 220 à la validation et 430 aux tests. Pour PH2, le jeu de données a été divisé en 160 images d’entraînement, 10 images de validation et 30 images de test. Une résolution d’image constante de 256 × 256 pixels était maintenue tout au long du pipeline de prétraitement.
Augmentation et normalisation des données
Des techniques d’augmentation de données ont été appliquées pour améliorer la généralisation des modèles. Des basculements horizontaux, verticaux et rotations aléatoires dans une plage de ±15° ont été mis en place. La correction gamma et la transformation logarithmique ont été appliquées avec une probabilité de 0,3 chacune. Les valeurs des pixels ont été normalisées à l’aide des statistiques ImageNet (moyenne = [0,485, 0,456, 0,406], écart-type = [0,229, 0,224, 0,225]).
Redimensionner toutes les images d’entrée à 256 × 256 pixels : Pour garantir l’efficacité de calcul, toutes les images d’entrée furent redimensionnées à une résolution uniforme de 256 × 256 pixels. Bien que les images haute résolution, comme celles du jeu de données (1000 × 1000 pixels), contiennent des détails fins et des informations de texture essentielles pour une délimitation précise des limites de lésions, augmenter la résolution n’a pas amélioré de manière significative les performances du modèle. Par conséquent, la résolution n’a pas été augmentée afin de maintenir un équilibre entre l’efficacité de calcul et la précision du modèle. Des travaux futurs pourraient approfondir les effets des entrées à plus haute résolution afin de déterminer si les bénéfices de la précision de segmentation justifient l’augmentation du coût de calcul. Les images étaient converties au format RVB lorsque cela était nécessaire. Les données prétraitées étaient sauvegardées dans des annuaires structurés tout en conservant les divisions originales du jeu de données.
Implémentation de l’architecture HMP-MUNet
Conception de l’architecture réseau
HMP-MUNet a été configuré selon la structure hiérarchique encodeur-décodeur en forme de U comme montré à la Figure 1. Le réseau était configuré avec une expansion progressive des canaux : 8→16→32→64→128→256 canaux à travers les niveaux de l’encodeur.
Un changement architectural significatif a été apporté en réduisant la profondeur du réseau de quatre niveaux hiérarchiques à deux, ce qui aide à simplifier le modèle et à améliorer l’efficacité de calcul. Bien que la profondeur soit réduite, les dimensions du canal ont été augmentées à chaque niveau, permettant au réseau de capturer des caractéristiques plus riches et plus expressives. Pour améliorer davantage l’apprentissage des caractéristiques du modèle, des mécanismes d’attention ont été introduits afin de concentrer le réseau sur les caractéristiques les plus pertinentes à plusieurs échelles. Ces mécanismes d’attention compensent efficacement la perte potentielle de l’abstraction hiérarchique des caractéristiques due à l’architecture plus superficielle.
L’encodeur a été initialisé avec des opérations doubles Conv2D pour l’extraction initiale des caractéristiques des tenseurs d’entrée X(C, H×W). La configuration alternée de trois modules spécialisés a été mise en œuvre : le schéma de commutation basé sur la vision de haute qualité (H-VSS), le réseau flexible parallèle multi-profondeur (PMFlex) et le réseau de fusion à attention dilatée multi-échelle (MSDAFN).
Le modèle utilise les statistiques ImageNet (moyenne et écart-type) pour la normalisation, évitant ainsi de recalculer pour chaque jeu de données, améliorant ainsi l’efficacité de calcul. Il garantit la stabilité et la généralisation, tirant parti de l’utilisation étendue d’ImageNet dans les tâches de vision par ordinateur. Ce choix simplifie la conception en réduisant la complexité du prétraitement et améliore la transférabilité entre ensembles de données, permettant au modèle de se généraliser efficacement sur diverses images de lésions cutanées.
Implémentation du schéma de commutation basé sur la mamba à vision d’ordre élevé (H-VSS)
Le module H-VSS a été implémenté selon l’architecture de la Figure 2. La normalisation des couches (LN) et l’activation de Hardswish (HS) ont été configurées avec des connexions résiduelles suivant l’équation (1) :

La composante Descripteur Spatial Local (LSD) a été mise en œuvre pour maintenir la cohérence spatiale. Le module de balayage spatial sélectif 2D (SS2D) a été configuré avec des schémas de balayage multidirectionnel suivant l’équation (2) :

Le traitement multi-couches par perceptron (MLP) a été ajouté avec des connexions résiduelles comme spécifié dans l’équation (3) :

Le mécanisme de balayage sélectif 2D d’ordre élevé (H-SS2D) projette des caractéristiques d’entrée dans un espace de dimension 2C pour une modélisation contextuelle améliorée.
Implémentation du réseau flexible parallèle multi-profondeur (PMFlex)
Le module PMFlex a été configuré selon les spécifications de la Figure 3 . La normalisation des couches a été appliquée aux cartes de caractéristiques d’entrée X(C, H×W), puis partitionnées en quatre segments le long de la dimension du canal suivant l’équation (4) :

Chaque fonctionnalité segmentée Y_i était traitée via des modules partagés Visual Mamba (VMamba). Les sorties traitées étaient concaténées, et un raffinement appliqué via la normalisation des couches et la projection comme décrit dans l’équation (5) :

Mise en œuvre du réseau de fusion à attention dilatée multi-échelle (MSDAFN)
Le module MSDAFN a été implémenté selon l’architecture Figure 4. Les opérations de convolution parallèles étaient configurées avec des taux de dilatation de 6, 12 et 18 pour l’extraction de caractéristiques à plusieurs échelles selon l’équation (6) :

Des fonctionnalités multi-échelles ont été intégrées par concaténation de canaux comme spécifié dans l’équation (7) :

Des mécanismes d’attention double ont été mis en place pour le recalibrage des fonctionnalités. Les pondérations d’attention des canaux ont été calculées à l’aide du regroupement global moyen suivant l’équation (8) :

La pondération de l’attention du canal a été appliquée comme décrit dans l’équation (9) :

L’attention spatiale a été configurée par des opérations de convolution suivant l’équation (10) :

L’attention du canal et spatiale a été combinée comme spécifié dans l’équation (11) :

Configuration et optimisation de l’entraînement
Configuration de l’environnement
L’environnement expérimental était configuré sur un système Ubuntu 20.04 avec un GPU (32 Go de VRAM). Python 3.8, un framework d’apprentissage profond (RRID : SCR_018536) et CUDA 11.8 ont été installés. La taille de l’image d’entrée était réglée à 256 × 256 pixels pour les tâches de lésions cutanées.
Fonction de perte et configuration de l’optimiseur
La fonction de perte BceDice a été implémentée pour l’optimisation de l’entraînement. L’optimiseur AdamW était configuré avec un taux d’apprentissage initial de 0,001, une taille de lot de 8 et 250 époques d’entraînement. Une décroissance de poids de 1 × 10⁻5 a été appliquée pour la régularisation.
Pour la configuration de base, cette étude se réfère aux travaux de Liu et al. (2024) sur le modèle Vmamba, qui a été utilisé comme modèle d’espace d’état visuel pour la tâche11 de classification médicale de l’image. Les configurations exactes et les scripts utilisés pour implémenter Vmamba sont basés sur leurs travaux publiés et adaptés pour cette étude, avec des ajustements spécifiques pour mieux s’adapter à l’ensemble de données d’imagerie médicale.
La planification des taux d’apprentissage a été mise en place par recuit cosinus avec des redémarrages à chaud. Configurez T_0 = 10 époques pour la période de redémarrage initiale, T_mult = 2 pour la multiplication des périodes, et η_min = 1 × 10⁻6 pour le taux d’apprentissage minimum.
Optimisation des hyperparamètres
Pour garantir la reproductibilité et la cohérence, toutes les expériences ont été menées avec une graine aléatoire fixe de 42. L’optimisation systématique des hyperparamètres a été réalisée en se concentrant sur la taille du lot, le taux d’apprentissage et le taux de chemin de chute. Les tailles de lots de 4, 8, 16 et 32 ont été évaluées. Les taux d’apprentissage de 0,0005, 0,001, 0,0015 et 0,002 ont été testés. La performance de validation a été surveillée en utilisant le coefficient de similarité de dés (DSC) comme métrique principale. Les tailles de lots supérieures à 8 peuvent entraîner un débordement de mémoire sur les GPU de moins de 32 Go de VRAM.
Évaluation du modèle et évaluation de la performance
Configuration des métriques d’évaluation
Des métriques d’évaluation complètes, incluant la Moyenne d’Intersection sur Union (mIoU), le Coefficient de Similarité de Dice (DSC), la Sensibilité (Sen), la Spécificité (Spe) et la Précision (Acc), ont été mises en œuvre. Les métriques ont été calculées selon les formulations suivantes :
L’intersection moyenne sur Union (mIoU) quantifie le chevauchement entre segmentation prédite et segmentation de vérité sur le terrain :

Le coefficient de similarité des dés (DSC) mesure la cohérence de segmentation. Les valeurs varient de 0 à 1, les valeurs plus élevées indiquant une meilleure performance :

La sensibilité (Sen) mesure la capacité du modèle à détecter des échantillons positifs :

La spécificité S(Spe) évalue la reconnaissance correcte négative de l’échantillon :

La précision (Acc) mesure la correction globale de la prédiction :

Les paramètres (M) reflètent la complexité du modèle, mesurant les paramètres totaux entraînables :

où Pi est le nombre de paramètres dans la i-ème couche, et N est le nombre total de couches. Des comptes de paramètres plus petits indiquent des modèles plus légers adaptés au déploiement clinique.
Protocole de l’étude d’ablation
Des études d’ablation complètes ont été menées pour valider les contributions des composants architecturaux selon la conception expérimentale du Tableau 1. Quatre variantes architecturales ont été évaluées : H-MUNet (base sans MSDAFN et PMFlex), HP-MUNet (sans MSDAFN), HM-MUNet (sans PMFlex) et HMP-MUNet (modèle complet).
Des paramètres d’entraînement identiques ont été configurés pour toutes les variantes : taux d’apprentissage 0,001, taille de lot 8, 250 époques. La convergence de l’entraînement a été surveillée et les améliorations de performance validées pour chaque ajout de composant.
Analyse de l’efficacité computationnelle
Les métriques d’efficacité computationnelle, incluant le nombre de paramètres, les FLOPs et le temps d’inférence entre différentes architectures, ont été mesurées. L’évaluation du temps d’inférence du modèle sur un GPU clinique standard montre que, bien qu’il performe mieux sur des GPU haute performance, le modèle est environ 70 % plus lent sur le matériel plus couramment utilisé. Néanmoins, il reste capable de vitesses d’inférence efficaces, ce qui le rend adapté à un déploiement clinique pratique. Surveillez l’utilisation de la mémoire GPU pendant l’entraînement pour garantir la stabilité du système. Mémoire GPU minimale recommandée pour la taille 8 du lot.
Validation et analyse
Benchmarking des performances
Les performances de HMP-MUNet ont été comparées à des méthodes de pointe sur les deux ensembles de données. Tous les modèles comparés ont été mis en œuvre et entraînés selon les mêmes protocoles de répartition des données, prétraitement, augmentation des données et entraînement afin de garantir que les différences de performance soient uniquement dues à des différences architecturales. Les résultats quantitatifs, incluant des améliorations DSC de 2,89 % et 5,25 % sur les ensembles de données ISIC2018 et PH2, ont été documentés respectivement. L’étude a validé que le nombre de paramètres est réduit de 4,55 fois par rapport à la référence U-Net.
Analyse statistique
Des tests de signification statistique ont été réalisés à l’aide de tests t par paires pour la comparaison de performances. Des intervalles de confiance ont été calculés pour les indicateurs rapportés. La reproductibilité a été assurée grâce à plusieurs entraînements avec différentes graines aléatoires.
Les configurations optimales d’hyperparamètres ont été enregistrées avec une taille de lot 8 et un taux d’apprentissage de 0,001, atteignant un DSC maximal de 0,9585 sur le jeu de données PH2 et 0,9044 sur le jeu de données ISIC2018, comme documenté dans les résultats expérimentaux. Des données de validation suffisantes ont été assurées pour éviter le sur-ajustement. Les courbes de perte de validation ont été surveillées pour appliquer des critères d’arrêt précoce.