Simple Science

Scienza all'avanguardia spiegata semplicemente

# Informatica# Visione artificiale e riconoscimento di modelli

Sviluppi nelle Tecniche di Segmentazione delle Immagini Mediche

Un nuovo metodo migliora l'accuratezza della segmentazione delle immagini nella sanità.

― 5 leggere min


Nuove tecnicheNuove tecnichenell'imaging medicoaffrontano le sfide dei dati.Metodi di segmentazione migliorati
Indice

La segmentazione delle immagini mediche è un compito importante che implica identificare diverse parti delle immagini mediche, come organi e tumori. Tradizionalmente, i modelli di deep learning usati per questo compito necessitano di grandi quantità di dati etichettati provenienti dalla stessa fonte per funzionare bene. Tuttavia, nella realtà, può esserci un gap significativo tra i dati usati per l'allenamento e quelli usati per il test. Questo gap spesso si verifica perché le immagini mediche possono essere scattate in modi o formati diversi, rendendo difficile per i modelli performare in modo accurato.

La sfida dello spostamento di dominio

In molti casi, i dati di addestramento possono venire da un tipo di imaging medico, come la risonanza magnetica, e i dati di test da un altro, come le TC. Questa situazione crea un problema chiamato spostamento di dominio. Quando c'è uno spostamento di dominio, il modello tende a fare fatica e le sue prestazioni calano significativamente.

Una soluzione a questo problema è etichettare manualmente alcuni dati dal dominio target. Tuttavia, farlo richiede molto tempo e costo, coinvolgendo dottori o radiologi esperti. Per superare questo problema, i ricercatori hanno iniziato a esplorare nuovi metodi per trasferire conoscenze da immagini etichettate a quelle non etichettate, specialmente in modi che non richiedono accesso ai dati di origine.

Adattamento al dominio senza sorgente (SFDA)

L'adattamento al dominio senza sorgente (SFDA) è un nuovo approccio che mira a risolvere il problema di segmentare le immagini quando non hai accesso ai dati originali. In questo metodo, i ricercatori usano un modello pre-addestrato, che è stato addestrato su immagini etichettate del dominio sorgente e lo adattano per lavorare su immagini non etichettate del dominio target.

Il metodo SFDA consiste in due fasi principali: la prima fase si concentra sull'allineamento delle caratteristiche con i prototipi di classe, e la seconda fase usa l'Apprendimento Contrastivo per migliorare le prestazioni del modello.

Fase 1: Allineamento delle caratteristiche ancorate ai prototipi

Nella prima fase, i ricercatori usano il modello addestrato per creare prototipi per ogni classe. Questi prototipi rappresentano le caratteristiche chiave di ogni classe apprese dai dati di origine. Poiché non possiamo accedere ai dati originali, ci affidiamo a questi prototipi per aiutare ad allineare le caratteristiche nelle immagini target.

Per fare questo, il metodo include un processo chiamato costo di trasporto, che misura quanto le caratteristiche target possano essere spostate verso questi prototipi. L'obiettivo è assicurarsi che le caratteristiche delle immagini target siano il più vicino possibile ai loro prototipi corrispondenti.

Tuttavia, se ci concentriamo solo su questo allineamento, il modello potrebbe comunque risultare inclinato verso una singola classe, il che significa che può etichettare erroneamente molti pixel. Per evitare ciò, viene aggiunto un meccanismo di trasporto inverso. Questo assicura che tutti i prototipi possano attribuire alcune caratteristiche dalle immagini target, aiutando a mantenere la diversità nelle previsioni.

Fase 2: Apprendimento contrastivo con previsioni inaffidabili

Una volta allineate le caratteristiche con i prototipi, la seconda fase utilizza l'apprendimento contrastivo. Questa fase aiuta il modello a diventare più robusto usando pixel che inizialmente avevano previsioni inaffidabili.

In questo processo, il modello esamina le previsioni fatte e identifica quali pixel sono incerti. Questi pixel incerti possono aiutare a rifinire il processo di apprendimento. Invece di ignorare queste previsioni, il modello le usa per creare una distinzione più netta tra le classi.

Per ogni classe, il modello separa i pixel in campioni "query", che probabilmente sono accurati, e campioni "negativi", che probabilmente sono errati. Poi, usa queste informazioni per garantire una migliore distribuzione compatta delle caratteristiche, portando a un miglior processo decisionale.

Sperimentazione e risultati

Per testare l'efficacia di questo approccio a due fasi, i ricercatori hanno condotto esperimenti su un compito specifico riguardante gli organi addominali. Hanno raccolto risonanze magnetiche e TC da fonti diverse per simulare le condizioni viste negli scenari reali.

I risultati hanno mostrato che il metodo ha superato le tecniche all'avanguardia esistenti. Ha ottenuto una maggiore accuratezza nella segmentazione degli organi, come il fegato e i reni, anche nei casi in cui c'era un grande gap tra i tipi di dati di addestramento e test.

Inoltre, i risultati hanno indicato che il processo a due fasi è stato vantaggioso. Quando i ricercatori hanno rimosso either l'allineamento delle caratteristiche o la fase di apprendimento contrastivo, le prestazioni del modello sono diminuite significativamente. Questo dimostra che entrambe le fasi sono importanti per ottenere i migliori risultati.

Conclusione

In sintesi, il nuovo framework di adattamento al dominio senza sorgente offre una soluzione promettente alle sfide viste nella segmentazione delle immagini mediche quando i dati etichettati del dominio sorgente non sono disponibili. Utilizzando un processo a due fasi che combina l'allineamento delle caratteristiche con le informazioni sui prototipi e l'apprendimento contrastivo usando previsioni inaffidabili, il modello può adattarsi efficacemente a diverse modalità di imaging medico.

Questi progressi non solo migliorano le prestazioni del modello, ma riducono anche la dipendenza da costose annotazioni esperte. Questo lavoro apre la strada a applicazioni più efficaci e pratiche del machine learning nel campo medico, contribuendo ad aumentare l'accuratezza delle diagnosi e a migliorare la cura dei pazienti senza la necessità di risorse estese.

Le ricerche future potrebbero riguardare il perfezionamento ulteriore di questi metodi ed esplorare la loro applicabilità in altri domini medici. La speranza è che con modelli più robusti, il processo di diagnosi delle malattie tramite analisi delle immagini mediche possa diventare più veloce, più efficiente e più accessibile per i fornitori di assistenza sanitaria.

Fonte originale

Titolo: Source-Free Domain Adaptation for Medical Image Segmentation via Prototype-Anchored Feature Alignment and Contrastive Learning

Estratto: Unsupervised domain adaptation (UDA) has increasingly gained interests for its capacity to transfer the knowledge learned from a labeled source domain to an unlabeled target domain. However, typical UDA methods require concurrent access to both the source and target domain data, which largely limits its application in medical scenarios where source data is often unavailable due to privacy concern. To tackle the source data-absent problem, we present a novel two-stage source-free domain adaptation (SFDA) framework for medical image segmentation, where only a well-trained source segmentation model and unlabeled target data are available during domain adaptation. Specifically, in the prototype-anchored feature alignment stage, we first utilize the weights of the pre-trained pixel-wise classifier as source prototypes, which preserve the information of source features. Then, we introduce the bi-directional transport to align the target features with class prototypes by minimizing its expected cost. On top of that, a contrastive learning stage is further devised to utilize those pixels with unreliable predictions for a more compact target feature distribution. Extensive experiments on a cross-modality medical segmentation task demonstrate the superiority of our method in large domain discrepancy settings compared with the state-of-the-art SFDA approaches and even some UDA methods. Code is available at https://github.com/CSCYQJ/MICCAI23-ProtoContra-SFDA.

Autori: Qinji Yu, Nan Xi, Junsong Yuan, Ziyu Zhou, Kang Dang, Xiaowei Ding

Ultimo aggiornamento: 2023-07-19 00:00:00

Lingua: English

URL di origine: https://arxiv.org/abs/2307.09769

Fonte PDF: https://arxiv.org/pdf/2307.09769

Licenza: https://creativecommons.org/licenses/by/4.0/

Modifiche: Questa sintesi è stata creata con l'assistenza di AI e potrebbe presentare delle imprecisioni. Per informazioni accurate, consultare i documenti originali collegati qui.

Si ringrazia arxiv per l'utilizzo della sua interoperabilità ad accesso aperto.

Altro dagli autori

Articoli simili