Supervised Machine-Learning Enables Segmentation and Evaluation of Heterogeneous Post-treatment Changes in Multi-Parametric MRI of Soft-Tissue Sarcoma

Frontiers in Oncology 2019 AI 8 Explanations View Original
Original Paper (PDF)

Unable to display PDF. Download it here or view on PMC.

Plain-English Explanations
Pages 1-2
Why Tumor Heterogeneity Makes STS Response Assessment So Difficult

Soft-tissue sarcoma (STS) is a rare malignancy arising in connective tissues, with approximately 3,300 new cases diagnosed annually in the UK alone. The 5-year survival rate sits at roughly 53%, and a central reason for this sobering figure is the challenge of accurately assessing treatment response. Unlike many carcinomas, STS tumors are biologically and spatially heterogeneous, often containing a mixture of viable cellular tumor, fat, necrosis, and cystic tissue regions within the same lesion. Standard size-based response criteria (such as RECIST) are poorly suited to tumors that may shrink only minimally even when treatment is working, because partial responses can manifest as changes in internal tumor composition rather than overall volume.

The multi-parametric MRI opportunity: Quantitative MRI offers a non-invasive window into the biology of STS tumors. Diffusion-weighted MRI (DW-MRI) produces maps of apparent diffusion coefficient (ADC), which inversely reflects tissue cellularity. Dynamic contrast-enhanced MRI and Dixon fat-water imaging add complementary dimensions, capturing vascular enhancement and fat content, respectively. When these sequences are combined into a multi-parametric protocol, the resulting data is rich enough to theoretically identify each tissue subcomponent. The clinical problem is that manually interpreting these multi-dimensional maps across the entire tumor volume is labor-intensive, subjective, and practically infeasible at scale.

The machine-learning hypothesis: This 2019 pilot study from the Royal Marsden Hospital and the Institute of Cancer Research, London, asks whether supervised machine-learning (ML) classifiers can automate the segmentation of these tissue subcomponents. If so, quantitative habitat maps could be generated rapidly from any multi-parametric MRI scan, enabling clinicians to track not just whether a tumor is responding, but which compartments are changing and by how much. This addresses a fundamental limitation of conventional whole-tumor average metrics, which can mask important localized changes within heterogeneous masses.

The study is framed as a technical development investigation, prospectively enrolling 18 patients with retroperitoneal sarcoma and testing eight different ML classifiers head-to-head, with a subset of eight patients providing paired pre- and post-radiotherapy scans to demonstrate clinical utility.

TL;DR: STS affects roughly 3,300 UK patients per year with a 53% five-year survival rate. Standard size-based response criteria fail for heterogeneous tumors. This pilot study tests eight ML classifiers on multi-parametric MRI (ADC, fat-fraction, enhancement-fraction) from 18 retroperitoneal sarcoma patients, with 8 patients providing pre- and post-radiotherapy pairs, aiming to automate tissue subtype segmentation for response monitoring.
Pages 2-3
Study Design, Imaging Protocol, and Tissue Classification Framework

Patient cohort: Eighteen patients with retroperitoneal sarcomas were enrolled in this prospective single-center study (11 male, 7 female; age range 43-76 years). The cohort included 14 liposarcomas, 2 leiomyosarcomas, 1 spindle cell sarcoma, and 1 synovial sarcoma, reflecting the typical histological distribution of retroperitoneal STS. All participants underwent baseline MRI, and 8 of the 18 who received pre-operative radiotherapy underwent a second MRI 2-4 weeks after completing treatment. Ethical approval was obtained from the National Research Ethics Service (NRES) Cambridge East committee (REC: 13/EE/1086), and all patients provided written informed consent.

Imaging protocol: All scans were acquired on a 1.5 T Siemens MAGNETOM Aera scanner using anterior body matrix and posterior spine matrix receive coils. Following anatomical T1- and T2-weighted sequences, functional imaging included: diffusion-weighted imaging (DWI) for ADC maps, Dixon imaging for fat-fraction (FF) maps, and pre- and post-gadolinium (Gd) T1-weighted imaging for enhancement-fraction (EF) maps. The field of view covered the full tumor extent. Maps were rescaled to a uniform [0,1] range: ADC was normalized by dividing by 3 x 10^-3 mm2/s, and EF was clipped to [0,1]. Tumor volumes of interest (VOIs) were defined by an expert radiologist with 16 years of experience, who outlined the entire tumor on axial T2-weighted images; these VOIs were then propagated to all quantitative parameter maps.

Tissue habitat definitions: The authors defined four biologically motivated tissue classes using a decision tree. Class 1 (enhancing viable tumor, low ADC, high EF) and Class 2 (non-enhancing viable tumor, low ADC, low EF) represent cellular tumor with and without active vascularity. Class 3 (necrotic or cystic tissue, high ADC) captures therapy-induced cell death and fluid-filled regions. Class 4 (fat, high FF) represents fatty tissue components common in liposarcoma. A fifth "novelty" class was added to capture voxels whose MRI signature fell outside the training data distribution, allowing the classifier to flag uncertain regions rather than force them into an incorrect category. Expert radiologists drew small 2 cm2 training regions of interest within each class to generate labeled ground truth data for model training.

The three quantitative MRI features (ADC, FF, EF) form the input feature vector for each voxel. The low dimensionality of this feature space (three parameters per voxel) is important because it means simpler ML methods may perform as well as complex ones, and training can be accomplished rapidly on standard hardware.

TL;DR: 18 retroperitoneal STS patients (14 liposarcomas) scanned at 1.5 T; 8 with paired pre- and post-radiotherapy MRI. Three quantitative MRI features per voxel: ADC (cellularity), fat-fraction (fatty content), enhancement-fraction (vascularity). Four tissue classes defined by expert-drawn 2 cm2 training ROIs, plus a novelty class for out-of-distribution voxels. NRES ethics approved.
Pages 3-4
Eight Classifiers Benchmarked: From Logistic Regression to Neural Networks

The study evaluated eight supervised ML classifiers using the Scikit-Learn Python library. The methods spanned a wide range of algorithmic families: Logistic Regression (LR) as a linear baseline; Support Vector Machine (SVM) with a radial basis function kernel; Random Forest (RF), an ensemble of decision trees; k-Nearest Neighbor (kNN), classifying voxels by majority vote among the k most similar training samples; Kernel Density Estimation (KDE), fitting a probabilistic density to each class; Naïve-Bayes (NB), applying Bayes' theorem under a class-conditional independence assumption; and a Neural Network (NN) with 20 nodes and three fully connected layers. A variant of KDE with hyperparameter optimization (KDE-opt) was also tested.

Hyperparameter optimization: Each classifier was tuned via an exhaustive cross-validation approach, sweeping through the relevant hyperparameter range for each method. For LR, the inverse regularization strength C was swept from 10^-3 to 10^10. For SVM, the penalty parameter C was varied from 10^-3 to 10^5. For kNN, the number of neighbors ranged from 1 to 100. For KDE, bandwidth was tested from 0.01 to 10. The NN used a fixed architecture but was evaluated across different configurations. The optimal hyperparameter for each method was selected as the value producing the highest median cross-validation accuracy on the labeled training data.

Markov Random Field (MRF) post-processing: A key methodological contribution of the paper is the incorporation of a Markov Random Field prior distribution into the classification pipeline. Raw voxel-wise classification produces noisy habitat maps because neighboring voxels are classified independently, without any spatial coherence constraint. The MRF introduces a spatial prior that penalizes abrupt changes in tissue class between adjacent voxels, effectively smoothing the classification result while still being driven by the trained classifier's probability outputs. This is a well-established technique in image segmentation (analogous to conditional random fields) that significantly improves the visual quality and anatomical plausibility of the resulting maps.

Computational performance: Training and prediction times were measured on a standard 3.5 GHz personal machine with 16 GB RAM and an Intel Iris Plus graphics card, using 1,350-3,000 training samples and 135-300 validation samples. NB achieved median training time of 0.73 ms and prediction time of 0.69 ms, making it by far the fastest method. SVM was the slowest, with median training time of 208.69 ms (range 96.01-1,745.5 ms) and prediction time of 12.94 ms. These computational differences become clinically relevant when classifying millions of voxels across a whole tumor volume.

TL;DR: Eight classifiers compared: LR, SVM, RF, kNN, KDE, KDE-opt, Naïve-Bayes, and a 20-node 3-layer NN. All optimized via exhaustive cross-validation. Naïve-Bayes trained in 0.73 ms vs. SVM at 208.69 ms median. MRF post-processing added spatial coherence to raw voxel-wise predictions, substantially smoothing habitat maps. All implemented in Scikit-Learn.
Pages 4-6
Cross-Validation Accuracy: Five Methods Perform Equivalently at 82%

The headline result is that 5 of the 8 ML techniques achieved statistically indistinguishable cross-validation accuracies, with a median of 82.2% (range 80.5-82.5%) when classifying voxels across all five tissue classes simultaneously. These five methods were LR, RF, kNN, KDE-opt, and NB. The three remaining methods (SVM, KDE, and NN) underperformed relative to this cluster, though the differences were modest. The relative insensitivity of performance to the choice of ML algorithm, provided the method has sufficient flexibility, is a direct consequence of the low-dimensional input space: with only three features per voxel, most classifiers can adequately model the decision boundaries separating the tissue classes.

Per-class accuracy: The cross-validation accuracy varied across the five tissue classes. NB scored highest for Class 3 (necrotic/cystic tissue, high ADC) and Class 4 (fatty tissue, high FF), while kNN scored highest for distinguishing Class 1 (enhancing viable tumor) from Class 2 (non-enhancing viable tumor). This per-class variation is clinically meaningful: the most challenging discrimination is between enhancing and non-enhancing cellular tumor (both with low ADC), which requires the enhancement-fraction feature to provide the decisive signal. Classes 3 and 4 are more separable because their ADC and fat-fraction values are distinctly different from cellular tumor regions.

MRF improvement: When the MRF prior was applied to the NB classifier output on the held-out test patient (not included in any training data), the resulting habitat maps showed substantially reduced classification noise. The MRF progressively converged to a smooth segmentation through iterative updates, with the convergent properties of the algorithm demonstrated visually in the paper. "Spie charts" (a variant of pie charts where segment radii encode the mean ADC of each tissue sub-compartment and angles encode volume proportions) were used to visualize the tumor composition compactly, enabling at-a-glance comparison of tissue distributions before and after treatment.

Why Naïve-Bayes was selected: Given equivalent accuracy among the top five methods, the authors selected NB for downstream application due to its exceptional computational efficiency. At 0.73 ms training time and 0.69 ms prediction time on a standard personal computer, NB enables rapid whole-tumor classification that is practical for clinical implementation without specialized computing hardware. This reflects a pragmatic principle: when multiple algorithms perform equivalently, clinical deployment considerations (speed, interpretability, reproducibility) should guide the final choice.

TL;DR: 5 of 8 ML methods achieved 82.2% median cross-validation accuracy (range 80.5-82.5%) with no statistically significant differences between them. NB selected as final method due to 0.73 ms training and 0.69 ms prediction times. MRF post-processing substantially reduced voxel-wise classification noise. kNN best for distinguishing enhancing vs. non-enhancing viable tumor; NB best for necrosis and fat classes.
Pages 6-7
Tracking Radiotherapy Response Through Habitat Maps in Three Patients

The most clinically relevant section of the paper applies the NB-MRF classifier to the eight patients who underwent paired pre- and post-radiotherapy MRI. Three representative cases are presented in detail, illustrating the range of treatment responses that multi-parametric habitat mapping can capture. These cases demonstrate that the method can detect changes in tumor composition that would not be visible using conventional size measurements or whole-tumor average MRI metrics.

Patient 1 (liposarcoma, no size response but compositional changes visible): Pre-treatment imaging showed a tumor dominated by enhancing viable tumor (Class 1, high vascularity) with a necrotic core (Class 3). Following radiotherapy, overall tumor volume did not change substantially, but the habitat map revealed a shift: the enhancing cellular compartment decreased while non-enhancing tumor (Class 2) and necrotic regions expanded. The Spie chart representation made this compositional redistribution immediately apparent, with the mean ADC of the surviving cellular compartments also increasing post-treatment, consistent with reduced cellularity.

Patient 2 (treatment-induced necrosis dominant): This patient demonstrated a more dramatic response, with a marked increase in the necrotic/cystic compartment (Class 3) following radiotherapy. The pre-treatment tumor contained predominantly viable cellular tumor, while the post-treatment map showed extensive replacement by high-ADC regions, indicating successful tumor cell kill. The ADC of the residual cellular regions also increased, suggesting reduced cellularity even within the surviving viable compartment. This type of compositional shift is precisely what conventional RECIST would miss if total tumor dimensions did not change.

Patient 3 (fat-containing liposarcoma with stable composition): The third case showed a tumor with a high proportion of fatty tissue (Class 4) at baseline, consistent with well-differentiated liposarcoma. Following treatment, the overall composition changed relatively little, with the fat component remaining the dominant tissue type. This case illustrates that the method is not biased toward detecting change when little has occurred, and that stable habitat distributions can be an informative finding in themselves, potentially indicating that the radiotherapy did not substantially alter the tumor's biology.

TL;DR: Three paired pre/post-radiotherapy cases demonstrated habitat map utility. Patient 1 showed compositional shift from enhancing to non-enhancing tumor without volume change. Patient 2 showed extensive necrosis induction with rising ADC in residual cellular regions. Patient 3 (fat-dominant liposarcoma) showed stable composition post-treatment. All three cases illustrated information beyond what RECIST size criteria would capture.
Pages 7-8
Interpreting the Findings: Low Dimensionality, Spatial Priors, and Clinical Potential

The authors provide a thoughtful discussion of why the five top-performing ML methods achieved statistically equivalent accuracy. The key insight is that with only three input features per voxel (ADC, FF, EF), the decision boundaries separating tissue classes are relatively simple and can be approximated equally well by methods as algorithmically different as logistic regression and random forest. In higher-dimensional settings (for example, if radiomic texture features or additional MRI contrasts were added), greater differences between methods would be expected. This finding has practical implications: it suggests that simple, interpretable ML methods can be deployed for this specific task without sacrificing accuracy, avoiding the need for computationally expensive deep learning pipelines.

The MRF contribution: The authors discuss the MRF smoothing step as an important practical contribution. Medical image segmentation produces spatially coherent regions, but voxel-wise classifiers applied independently do not inherently respect spatial structure. The MRF prior penalizes abrupt label transitions between neighboring voxels, which reflects the known biology of tumor tissue where gradual spatial transitions between compartments are far more common than sharp checkerboard patterns. The authors note that MRF-based smoothing is generalizable to any classification task in medical imaging, not just the STS application described here.

Comparison to prior work: Earlier studies had applied ML (including artificial neural networks) to DCE-MRI alone for sarcoma segmentation. This study advances the field by incorporating multi-parametric MRI, arguing that the additional information from DWI and Dixon sequences is necessary to reliably distinguish cellular tumor subtypes (enhancing vs. non-enhancing) from necrotic and fatty compartments. Single-modality DCE-MRI cannot discriminate fat from enhancing cellular tissue, for example, because both can show high signal. The multi-parametric approach resolves these ambiguities by leveraging complementary biological signals.

Clinical interpretation of ADC changes: The authors note that post-radiotherapy increases in ADC within the residual cellular compartments are consistent with reduced cellular packing following treatment-induced apoptosis or necrosis. Monitoring ADC changes within specific tissue compartments (rather than averaging across the whole tumor) provides a more sensitive readout of biological response, because the dilution effect of stable fat or necrosis on whole-tumor average ADC is eliminated by the compartment-specific analysis.

TL;DR: Low-dimensional input (3 features) explains why 5 diverse ML methods performed equivalently. MRF spatial prior is broadly applicable and reduces voxel-wise noise. Multi-parametric MRI outperforms single-modality DCE-MRI alone by resolving fat-vs-enhancement ambiguities. Compartment-specific ADC tracking is more sensitive to treatment response than whole-tumor average ADC.
Pages 8-9
Pilot-Scale Constraints and Barriers to Broader Validation

Small cohort and single center: The most significant limitation acknowledged by the authors is the small sample size of 18 patients, with only 8 providing paired pre- and post-treatment scans. At this scale, it is not possible to draw statistically robust conclusions about the association between habitat changes and clinical outcomes such as pathological response, local recurrence, or overall survival. The cohort is also histologically biased toward liposarcoma (14/18 cases), which may mean the classifier performs differently in other STS subtypes where the fatty tissue compartment is less prominent and the MRI appearance differs substantially.

Single-institution MRI protocol: All scans were acquired on a single 1.5 T Siemens scanner using a standardized protocol. The performance of the trained classifiers on scans acquired on different scanner manufacturers (GE, Philips), different field strengths (3 T), or with protocol variations (different b-values for DWI, different contrast agents) has not been tested. ADC values and fat-fraction estimates are known to vary across MRI platforms, which means a classifier trained at one institution may not generalize to another without recalibration. This is a persistent challenge in quantitative MRI research.

Expert-annotated training ROIs: The tissue class labels used for training were drawn by expert radiologists, making the method's accuracy contingent on the quality and consistency of expert annotation. Inter-observer variability in defining training ROIs has not been quantified in this study. If different expert annotators define the boundaries of each tissue class differently, the trained classifier will inherit this variability. Larger studies would benefit from multi-reader annotation protocols and inter-observer reliability reporting.

No histopathological correlation: The tissue classes (cellular tumor, necrosis, fat) are defined based on MRI parameters and expert radiological interpretation, but are not directly validated against histopathology specimens from the same tumor regions. Ideally, MRI-defined habitats would be correlated with spatially matched tissue samples, confirming that regions classified as necrotic by the algorithm correspond to histologically confirmed necrosis. Without this correlation, the biological interpretation of each class remains assumed rather than proven.

TL;DR: Key limitations: 18 patients total, 8 with paired scans, 14/18 liposarcomas. Single 1.5 T Siemens scanner with no cross-platform validation. Training labels from expert ROIs with unquantified inter-observer variability. No histopathological validation of MRI-defined tissue classes. Results need replication in larger, multi-center, multi-histology cohorts.
Pages 9-10
Scaling Up: Larger Cohorts, Deeper Architectures, and Outcome Correlations

Larger and multi-histology validation: The authors explicitly call for expanded studies enrolling a broader range of STS subtypes beyond retroperitoneal liposarcoma. This includes extremity sarcomas, which are anatomically more accessible and easier to resect, providing surgical specimens for histopathological correlation. Multi-center studies would be particularly valuable for testing whether classifiers trained on Marsden data generalize across scanner platforms and imaging protocols, a prerequisite for any clinical translation. Consortia with harmonized multi-parametric MRI protocols would enable pooling of data across institutions.

Integration with deep learning: The authors anticipate that modern deep learning architectures will eventually supersede the classical ML methods evaluated here. Fully convolutional networks and U-Net-style architectures can incorporate both voxel-level features and spatial context simultaneously, potentially outperforming the separate classification-plus-MRF pipeline used in this paper. Self-supervised and weakly supervised approaches could reduce the annotation burden associated with drawing expert training ROIs for each patient. The authors note that advances in AI are expected to significantly improve automatic segmentation accuracy in the coming years.

Outcome prediction: The immediate next step highlighted by the authors is correlating habitat map changes with clinical endpoints, including pathological tumor response at surgery, local recurrence rates, and disease-free survival. If specific habitat signatures at baseline or after radiotherapy reliably predict which patients will achieve complete pathological response at resection, these imaging biomarkers could inform adaptive treatment strategies. For example, patients showing early habitat changes consistent with treatment response might be candidates for surgical resection earlier than planned, while non-responders could be escalated to systemic therapy or higher radiotherapy doses.

Whole-body and multi-lesion extension: The current method is applied to individual tumors on a per-lesion basis. An extension to whole-body DW-MRI would enable multi-lesion assessment in patients with metastatic STS, providing a systemic view of treatment heterogeneity across all tumor sites simultaneously. This aligns with emerging approaches to whole-body MRI for oncological staging and response assessment, where the total burden of disease and the heterogeneity of response across lesions are recognized as important clinical variables.

TL;DR: Future directions include multi-center multi-histology cohorts for platform generalizability testing, deep learning (U-Net and fully convolutional architectures) to replace the classical ML-plus-MRF pipeline, correlation of habitat changes with pathological response and survival endpoints, and extension to whole-body DW-MRI for multi-lesion response assessment in metastatic STS.