Toolkit pentru modelarea CT pulmonară

Tip de date de intrare:

1. Preprocesare și segmentare 3D

Arhivă: Lung Cancer CT

Acest toolkit are rolul de a realiza automat filtrarea secțiunilor sănătoase 2D și 3D și segmentarea tumorilor de plămân de diferite dimensiuni din imagini CT (în format RAS - dreapta, anterior, posterior )/PT/NPY . Sistemul are la bază un model cascadat care stabilește dacă o secțiune 2D este sănătoasă, sau împarte un CT 3D în secțiuni 2D și le filtrează. În cazul 2D, secțiunile identificate drept tumorale sunt segmentate automat prin mai multe metode și este oferită utilizatorului masca obținută de ansamblu. În cazul 3D, se aplică algoritmi 2D secțiune cu secțiune și algoritmi 3D al căror rezultat este condiționat cu zerouri în zona considerată sănătoasă de algoritmii 2D.

CT-urile sunt preprocesate printr-o serie de operațiune de netezire și simplificare a distribuției valorilor:

  • Pe baza analizei exploratorii detaliate în [1], folosind datele de antrenare din setul Medical Segmentation Decathlon (MSD) a fost stabilită o fereastră 2D ce încadrează tumorile pulmonare mai exact decât întreg spațiul de 512x512 pixeli ai unei imagini CT (Fig. 1)
    Decatlon Medical - proiectia în 2D a tumorilor

    Fig. 1. Decatlon Medical - proiectia în 2D a tumorilor

  • Informațiile exterioare acestei ferestre sunt eliminate (Fig. 2)
    Filtrarea unei secțiuni CT pe baza ferestrei 2D

    Fig. 2. Filtrarea unei secțiuni CT pe baza ferestrei 2D

  • Pentru a crește generalitatea pacienților la antrenare, rotații de 15° au fost aplicate pe axa y în antrenarea modelului 3D (Fig. 3)
  • Secțiunea CT originală fără rotație
    (A)
    Secțiunea CT după rotație de 15° pe axa y
    (B)

    Fig. 3. Augmentarea datelor prin rotație de ±15° pe axa y

  • Au fost aplicate mai multe scheme de normalizare (Min-Max, Z-score, CLAHE). Dintre acestea, Min-Max a fost determinată empiric drept cea mai robustă la antrenare

2. Exemple de rezultate

2.1 Zona abdominală

Modelele 3D ce primesc ca intrare volume CT holistice, ce depășesc regiunea toracal pulmonară fac confuzii în ceea ce privește existența tumorilor. Din cauza structurilor anatomice complexe, bogate semantic în variații de contrast și luminozitate, rețelele prezic des tumoră în regiuni mari.

Secțiune abdominală cu predicție realizată de Model 3D holistic
(A)
Secțiune abdominală cu predicție realizată de un Model 2.5D
(B)

Fig. 4. Predicție rețea 3D holistică (A) și 2.5D pe secțiuni mai mici(B)

2.2 Zona pulmonară cu tumori

Pierderea din acuratețea segmentării tumorii în momentul reducerii Fals Pozitivelor

Fig. 5. Pierderea din acuratețea segmentării tumorii în momentul reducerii Fals Pozitivelor

Principala problemă observată în modelarea 2.5 și 3D a fost echilibrarea între generarea de Fals Pozitive asupra structurilor sănătoase cu mult contrast, și diminuarea acestui efect care reduce performanța în zona pulmonară cu tumori.

2.2 Segmentarea 2D obținută cu ajutorul AI explicabil

Din acest motiv, am implementat o metodă de segmentare 2D bazată pe explicațiile din spatele deciziilor unei sarcini de clasificare pe care o vom detalia în următoarea secțiune.

3. Segmentare prin clasificare explicabilă

Arhivă: CT xAI

1. Instalare

Opțiunea 1 — pip într-un virtualenv

python -m venv .venv
source .venv/bin/activate # Linux/Mac
# .venv\Scripts\activate # Windows PowerShell
pip install -U pip
pip install -r requirements.txt

Opțiunea 2 — conda

conda create -n lung_project python=3.12 -y
conda activate lung_project
pip install -r requirements.txt

Bibliografie

[1] Chiru, C. M., Nan, M., Trăscău, M., Florea, A. M. (2024). Novel Data Adaptation Techniques for Enhanced Lung Cancer Detection in CT Scans. In: 2024 26th International Symposium on Symbolic and Numeric Algorithms for Scientific Computing (SYNASC) (pp. 182-190). IEEE. https://ieeexplore.ieee.org/document/10896406

[2] Dugăeşescu, A., Chiru, C. M., Nan, M., Trăscău, M., Florea, A. M. (2024). Explainable cancer segmentation through classification. In: 2024 IEEE 20th International Conference on Intelligent Computer Communication and Processing (ICCP) (pp. 1-8). IEEE. https://ieeexplore.ieee.org/document/10793024