Questa ricerca è stata condotta in conformità con le linee guida istituzionali per la ricerca computazionale e l'elaborazione dei dati. Non sono stati coinvolti soggetti umani né animali vertebrati in questo studio.
Preparazione e preelaborazione del dataset
Raccolta e organizzazione dei dataset
Sono state raccolte immagini dermoscopiche dai dataset PH2, ISIC2018 e ISIC2017, con link specifici forniti nella Tabella dei Materiali. Il dataset PH2 contiene 200 immagini ad alta risoluzione (1000 × 1000 pixel), ISIC2018 comprende 2.594 immagini con corrispondenti maschere di segmentazione e ISIC2017 2.150 immagini con maschere di segmentazione. I dati sono stati organizzati in set di addestramento, validazione e test seguendo protocolli standard, assicurando che le immagini siano in formati compatibili (ad esempio, .jpg, .png o .tiff) e includano le corrispondenti maschere di verità sul terreno in formato binario.
I dati sono stati organizzati in set di addestramento, validazione e test seguendo protocolli standard. I dataset sono stati suddivisi come segue: per ISIC2018, sono state assegnate 1.815 immagini per l'addestramento, 259 per la validazione e 520 per i test. Per ISIC2017, sono state assegnate 1.500 immagini per l'addestramento, 220 per la validazione e 430 per i test. Per PH2, il dataset è stato suddiviso in 160 immagini di addestramento, 10 immagini di validazione e 30 immagini di test. Una risoluzione d'immagine costante di 256 × 256 pixel è stata mantenuta durante tutta la pipeline di preprocessing.
Aumento e normalizzazione dei dati
Sono state applicate tecniche di aumento dei dati per migliorare la generalizzazione dei modelli. Furono implementati ribaltamenti orizzontali, vertici e rotazioni casuali all'interno di un intervallo di ±15°. Correzione gamma e trasformazione logaritmica sono state applicate con una probabilità di 0,3 ciascuna. I valori dei pixel sono stati normalizzati utilizzando le statistiche di ImageNet (media = [0,485, 0,456, 0,406], deviazione standard = [0,229, 0,224, 0,225]).
Ridimensionare tutte le immagini di input a 256 × 256 pixel: per garantire l'efficienza computazionale, tutte le immagini di input furono ridimensionate a una risoluzione uniforme di 256 × 256 pixel. Sebbene le immagini ad alta risoluzione, come quelle nel dataset (1000 × 1000 pixel), contengano dettagli a grana fine e informazioni sulle texture cruciali per una delineazione accurata dei confini delle lesioni, aumentare la risoluzione non ha migliorato significativamente le prestazioni del modello. Pertanto, la risoluzione non è stata aumentata per mantenere un equilibrio tra efficienza computazionale e accuratezza del modello. Lavori futuri potrebbero approfondire gli effetti degli input ad alta risoluzione per determinare se i benefici nell'accuratezza della segmentazione giustifichino l'aumento dei costi computazionali. Le immagini venivano convertite in formato RGB quando necessario. I dati preprocessati venivano salvati in directory strutturate mantenendo le suddivisioni originali del dataset.
Implementazione dell'architettura HMP-MUNet
Progettazione dell'architettura di rete
HMP-MUNet è stato configurato seguendo la struttura gerarchica a forma di U codificatore-decodificatore come mostrato nella Figura 1. La rete era configurata con espansione progressiva dei canali: 8→16→32→64→128→256 canali tra i livelli degli encoder.
Un cambiamento architettonico significativo è stato apportato riducendo la profondità della rete da quattro livelli gerarchici a due, il che aiuta a semplificare il modello e migliorare l'efficienza computazionale. Sebbene la profondità sia ridotta, le dimensioni del canale sono state aumentate a ogni livello, permettendo alla rete di catturare caratteristiche più ricche ed espressive. Per migliorare ulteriormente l'apprendimento delle caratteristiche del modello, sono stati introdotti meccanismi di attenzione per focalizzare la rete sulle caratteristiche più rilevanti su più scale. Questi meccanismi di attenzione compensano efficacemente la potenziale perdita di astrazione gerarchica delle caratteristiche dovuta all'architettura più superficiale.
L'encoder veniva inizializzato con operazioni doppie Conv2D per l'estrazione iniziale delle caratteristiche dai tensori di ingresso X(C, H×W). È stata implementata la configurazione alternata di tre moduli specializzati: High-order Vision Mamba-based Switching Scheme (H-VSS), Parallel Multi-depth Flexible Network (PMFlex) e Multi-Scale Dilated Attention Fusion Network (MSDAFN).
Il modello utilizza le statistiche ImageNet (media e deviazione standard) per la normalizzazione, evitando la necessità di ricalcolare per ogni dataset, migliorando così l'efficienza computazionale. Garantisce stabilità e generalizzabilità, sfruttando l'ampio utilizzo di ImageNet nei compiti di visione artificiale. Questa scelta semplifica il design riducendo la complessità del preprocessing e migliora la trasferibilità tra dataset, permettendo al modello di generalizzare efficacemente su diverse immagini di lesioni cutanee.
Implementazione dello schema di commutazione basato su mamba a visione di alto ordine (H-VSS)
Il modulo H-VSS è stato implementato secondo l'architettura della Figura 2. La normalizzazione dello strato (LN) e l'attivazione di Hardswish (HS) sono state configurate con connessioni residue seguendo l'Equazione (1):

La componente Local Spatial Descriptor (LSD) è stata implementata per mantenere la coerenza spaziale. Il modulo Scansione 2D Selettiva Spaziale (SS2D) è stato configurato con schemi di scansione multidirezionale seguendo l'Equazione (2):

Il processamento Multi-Layer Perceptron (MLP) è stato aggiunto con connessioni residue come specificato nell'Equazione (3):

Il meccanismo di scansione selettiva 2D di alto ordine (H-SS2D) proietta le caratteristiche di input nello spazio bidimensionale per una modellazione contestuale migliorata.
Implementazione della rete flessibile parallela multi-profondità (PMFlex)
Il modulo PMFlex è stato configurato secondo le specifiche della Figura 3 . La normalizzazione dei livelli è stata applicata alle mappe di caratteristiche di input X(C, H×W), poi partizionata in quattro segmenti lungo la dimensione del canale seguendo l'Equazione (4):

Ogni funzionalità segmentata Y_i veniva elaborata tramite moduli condivisi Visual Mamba (VMamba). Gli output processati venivano concatenati e il raffinamento veniva applicato tramite Normalizzazione e proiezione dello Strato, come descritto nell'Equazione (5):

Implementazione della rete di fusione a attenzione dilatata multiscala (MSDAFN)
Il modulo MSDAFN è stato implementato seguendo l'architettura Figure 4. Le operazioni di convoluzione parallele sono state configurate con velocità di dilatazione di 6, 12 e 18 per l'estrazione di caratteristiche multiscala secondo l'Equazione (6):

Le funzionalità multiscala sono state integrate tramite concatenazione di canali come specificato nell'Equazione (7):

Sono stati implementati meccanismi di doppia attenzione per la ricalibrazione delle caratteristiche. I pesi di attenzione dei canali sono stati calcolati utilizzando il pooling della media globale seguendo l'Equazione (8):

La ponderazione dell'attenzione del canale è stata applicata come descritto nell'Equazione (9):

L'attenzione spaziale è stata configurata tramite operazioni di convoluzione seguendo l'Equazione (10):

L'attenzione del canale e spaziale è stata combinata come specificato nell'Equazione (11):

Configurazione e ottimizzazione dell'addestramento
Configurazione dell'ambiente
L'ambiente sperimentale era configurato su un sistema Ubuntu 20.04 con una GPU (32 GB di VRAM). Sono stati installati Python 3.8, framework di deep learning (RRID: SCR_018536) e CUDA 11.8. La dimensione dell'immagine in input è stata impostata a 256 × 256 pixel per i compiti sulle lesioni cutanee.
Funzione di perdita e configurazione dell'ottimizzatore
La funzione di perdita BceDice è stata implementata per l'ottimizzazione dell'addestramento. L'ottimizzatore AdamW era configurato con un tasso di apprendimento iniziale di 0,001, una dimensione del lotto di 8 e 250 epoche di addestramento. Per la regolarizzazione fu applicato un decadimento del peso di 1 × 10⁻5 .
Per la configurazione di base, questo studio si riferisce al lavoro di Liu et al. (2024) sul modello Vmamba, utilizzato come modello di spazio di stato visivo per il compito11 di classificazione delle immagini mediche. Le configurazioni esatte e gli script utilizzati per implementare Vmamba si basano sul loro lavoro pubblicato e adattati per questo studio, con aggiustamenti specifici per adattarsi meglio al dataset di imaging medico.
La programmazione della velocità di apprendimento è stata impostata utilizzando il coseno con riavvii a caldo. Configurare T_0 = 10 epoche per il periodo di riavvio iniziale, T_mult = 2 per la moltiplicazione dei periodi e η_min = 1 × 10⁻6 per il tasso minimo di apprendimento.
Ottimizzazione degli iperparametri
Per garantire riproducibilità e coerenza, tutti gli esperimenti sono stati condotti utilizzando un seed casuale fisso di 42. L'ottimizzazione sistematica degli iperparametri è stata condotta concentrandosi sulla dimensione del lotto, la velocità di apprendimento e il tasso di drop path. Sono stati valutati i lotti di 4, 8, 16 e 32. Sono stati testati i tassi di apprendimento di 0,0005, 0,001, 0,0015 e 0,002. Le prestazioni di validazione sono state monitorate utilizzando il Coefficiente di Similarità dei Dados (DSC) come metrica primaria. Dimensioni di batch maggiori di 8 possono causare un overflow di memoria su GPU con meno di 32 GB di VRAM.
Valutazione del modello e valutazione delle prestazioni
Configurazione delle metriche di valutazione
Sono state implementate metriche di valutazione complete, tra cui Mean Intersection over Union (mIoU), Dice Similarity Coefficient (DSC), Sensitivity (Sen), Specificity (Spe) e Accuracy (Acc). Le metriche sono state calcolate secondo le seguenti formulazioni:
L'intersezione media sull'unione (mIoU) quantifica la sovrapposizione tra segmentazione di verità prevista e segmentazione di verità sul terreno:

Il Coefficiente di Similarità dei Dadi (DSC) misura la coerenza della segmentazione. I valori variano da 0 a 1, con valori più alti che indicano prestazioni migliori:

La sensibilità (Sen) misura la capacità del modello di rilevare campioni positivi:

La specificità S(Spe) valuta il corretto riconoscimento negativo del campione:

L'accuratezza (Acc) misura la correttezza complessiva della previsione:

I parametri (M) riflettono la complessità del modello, misurando i parametri totali addestrabili:

dove Pi è il numero di parametri nel i-esimo strato, e N è il numero totale di strati. Conteggi di parametri più piccoli indicano modelli più leggeri, adatti al dispiegamento clinico.
Protocollo dello studio dell'ablazione
Sono stati condotti studi completi di ablazione per validare i contributi dei componenti architettonici seguendo il disegno sperimentale nella Tabella 1. Sono state valutate quattro varianti architettoniche: H-MUNet (base senza MSDAFN e PMFlex), HP-MUNet (senza MSDAFN), HM-MUNet (senza PMFlex) e HMP-MUNet (modello completo).
Parametri di addestramento identici erano configurati per tutte le varianti: tasso di apprendimento 0,001, dimensione lotto 8, 250 epoche. La convergenza dell'addestramento è stata monitorata e i miglioramenti delle prestazioni sono stati validati per ogni aggiunta di componenti.
Analisi dell'efficienza computazionale
Sono state misurate le metriche di efficienza computazionale, inclusi il conteggio dei parametri, i FLOP e i tempi di inferenza tra diverse architetture. La valutazione del tempo di inferenza del modello su una GPU standard di livello clinico mostra che, sebbene abbia le migliori prestazioni su GPU ad alte prestazioni, il modello è circa il 70% più lento su hardware più comunemente utilizzato. Tuttavia, è ancora in grado di raggiungere velocità di inferenza efficienti, rendendolo adatto per un impiego clinico pratico. Monitorare l'uso della memoria GPU durante l'addestramento per garantire la stabilità del sistema. Memoria GPU minima consigliata per il batch 8.
Validazione e analisi
Benchmarking delle prestazioni
Le prestazioni di HMP-MUNet sono state confrontate con metodi all'avanguardia su entrambi i dataset. Tutti i modelli confrontati sono stati implementati e addestrati secondo gli stessi protocolli di suddivisione dei dati, preprocessing, aumento dei dati e addestramento per garantire che le differenze di prestazioni fossero dovute esclusivamente a differenze architettoniche. I risultati quantitativi, inclusi miglioramenti DSC del 2,89% e 5,25% rispettivamente sui dataset ISIC2018 e PH2, sono stati documentati. Lo studio ha convalidato che il conteggio dei parametri è ridotto di 4,55 volte rispetto alla base di U-Net.
Analisi statistica
I test di significatività statistica sono stati eseguiti utilizzando test t accoppiati per confronti delle prestazioni. Gli intervalli di confidenza sono stati calcolati per le metriche riportate. La riproducibilità è stata garantita attraverso molteplici sessioni di addestramento con semi casuali diversi.
Le configurazioni ottimali degli iperparametri sono state registrate con dimensione del lotto 8 e tasso di apprendimento 0,001, raggiungendo un picco di DSC di 0,9585 sul dataset PH2 e 0,9044 sul dataset ISIC2018, come documentato nei risultati sperimentali. Sono stati garantiti dati di validazione sufficienti per prevenire il sovrafitting. Le curve di perdita di validazione sono state monitorate per applicare criteri di stop precoce.