In vivo MRI based prostate cancer localization with random forests and auto-context model

Comput Med Imaging Graph 2016 Machine Learning 7 Explanations View Original
Original Paper (PDF)

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

Plain-English Explanations
Pages 1-3
The Challenge of Finding Cancer in the Entire Prostate

Prostate cancer is the most common non-skin cancer and second leading cause of cancer death in US men. Standard diagnosis relies on transrectal ultrasound (TRUS)-guided biopsy after elevated PSA, but this approach has well-documented problems: low specificity, frequent under-grading of tumor aggressiveness, and high rates of overdiagnosis and overtreatment.

Multiparametric MRI (mp-MRI) offers a powerful non-invasive alternative, combining T2-weighted imaging (structural anatomy), diffusion-weighted imaging (DWI, measuring water movement), and dynamic contrast-enhanced sequences. However, reading prostate MRI requires high radiologist expertise that is not widely available, and the large 3D image volumes are time-consuming to analyze. This is why MRI remains a second-line tool after repeated negative biopsies, rather than a first-line screening modality.

Computer-aided detection (CAD) systems could overcome these barriers, but most prior systems focus on cancer only in the peripheral zone (PZ) - the outer part of the prostate where cancer is most easily identified on MRI due to its distinctive texture. Yet prostate cancer also arises in the central gland (CG) and the transition zone (TZ). TZ cancer is particularly challenging to detect because this zone is the origin of benign prostatic hyperplasia (BPH), which creates a heterogeneous appearance that can mimic cancer.

Additionally, most existing CAD systems are semi-automatic - requiring a radiologist to manually outline suspicious regions before the computer classifies them as cancer or benign. This approach is labor-intensive, subject to human error, and misses small cancers that weren't delineated. The goal of this study was to develop a fully automatic method for whole-prostate cancer localization at the voxel level directly from mp-MRI.

TL;DR: Most existing prostate cancer detection systems only analyze the peripheral zone and require manual input, leaving the central and transition zones and small cancers underdiagnosed.
Pages 4-5
Study Data: 26 Patients with Three MRI Types and Surgical Validation

The study analyzed 26 patients who underwent preoperative multiparametric MRI (using a 3T Philips scanner with endorectal coil) followed by radical prostatectomy between March 2008 and March 2010. Patients had received no prior treatment. Three MRI sequences were used: T2-weighted (T2W), diffusion-weighted imaging (DWI), and the derived ADC map (dADC). Across the 26 patients, 42 tumors were present: 29 in the peripheral zone, 7 in the central gland, and 6 in the transition zone.

The ground truth for cancer location was established through a careful correlative review by an expert genitourinary pathologist (8 years experience) and a radiologist (9 years experience in prostate MRI). The pathologist sliced the removed prostates, identified all tumor foci larger than 5mm in diameter, and the radiologist manually traced the corresponding regions on the preoperative MRI images. This pathology-MRI correlation provides the most reliable possible reference standard.

All MRI images were resampled to uniform resolution (0.3125mm x 0.3125mm x 3mm) and aligned (registered) together using the FLIRT registration tool. A specialized two-step registration process aligned DWI to T2W images - first using the bladder boundary as an anchor, then refining with prostate tissue boundaries - to minimize misalignment between the different MRI types.

TL;DR: 26 patients underwent preoperative MRI followed by surgery, with cancer locations verified by expert pathologist-radiologist correlation to create a precise ground truth for algorithm training and testing.
Pages 4-7
The Algorithm: Random Forests with Self-Improving Auto-Context

The system treats cancer localization as a voxel-by-voxel classification problem: every point in the 3D MRI volume is independently evaluated and given a probability score for being cancerous. The method uses random forests - an ensemble machine learning approach where 20 individual decision trees vote on each voxel's classification. Using many trees reduces variance and avoids the overfitting problems of single decision trees.

Each voxel is characterized by its appearance features - specifically Haar-like features derived from its local intensity patterns across the T2, DWI, and dADC images. Haar-like features are computationally efficient comparisons of average intensities in neighboring rectangular regions, encoding information about local gradients, edges, and texture. For each voxel, 10,000 randomly selected Haar-like features are evaluated.

The key innovation is the auto-context model: after the initial random forest generates a probability map of likely cancer locations, this probability map itself becomes an additional input feature for the next round of classification. In the second and subsequent iterations, the algorithm not only considers local image appearance but also context features - how confident the previous prediction was in neighboring voxels. This encodes spatial consistency: if surrounding voxels are likely cancerous, the current voxel is more likely to be cancerous too.

This iterative refinement continues for up to 10 iterations, with the probability maps progressively becoming more precise and the detected cancer regions more accurately matching ground truth. The biggest improvement occurs in the second iteration when context features first become available. Performance stabilizes after 3-4 iterations, and 10 iterations are used to ensure convergence and accuracy.

TL;DR: Random forests classify each image voxel as cancerous or not, with an auto-context model iteratively refining the probability map by incorporating spatial context from previous predictions.
Pages 8-9
How Performance Is Measured: Section-Based Evaluation

The primary performance metric is the section-based evaluation (SBE), which mirrors clinical practice. In standard prostate biopsy, the prostate is divided into sections (typically 6 or 12) and a needle is placed in each. The SBE measures what fraction of sections are correctly classified as cancerous or non-cancerous by the algorithm compared to the expert ground truth.

Standard 6 or 12-section division creates sections that are too large for precise needle guidance - a single section may span cancer and non-cancer areas. This study introduces a finer section division scheme that adapts to prostate anatomy: 9 sections for each of the base and mid-gland regions, and 4 sections for the apex (22 sections total). This finer granularity reduces the number of biopsy needle placements needed while improving the chance of accurately targeting cancerous tissue.

Additional evaluation metrics include section-based sensitivity (fraction of cancer sections correctly identified) and specificity (fraction of non-cancer sections correctly classified as non-cancerous), as well as Dice ratio (measuring overlap between predicted and actual cancer regions) and ROC curve analysis at both the voxel level and section level.

TL;DR: The study uses a fine 22-section prostate division scheme mirroring clinical biopsy practice, measuring what fraction of cancerous sections the algorithm correctly identifies compared to expert ground truth.
Pages 10-13
Results: Accurate Detection Across All Prostate Zones

The proposed method achieved a section-based evaluation (SBE) of 87.1%, sensitivity of 91.5%, and specificity of 85.1% on the 26-patient dataset. The ROC analysis gave an AUC of 0.832 at the voxel level and 0.883 for the section-based ROC analysis - demonstrating strong performance at both fine-grained and clinically relevant levels of evaluation.

The method significantly outperformed all four comparison algorithms: Naive Bayes (SBE 68.5%, sensitivity 73.7%, specificity 65.9%), standard SVM (SBE 71.4%, sensitivity 72.6%, specificity 71.0%), SVM with radial basis function kernel (SBE 79.9%, sensitivity 66.1%, specificity 83.8%), and AdaBoost (SBE 68.0%, sensitivity 94.6%, specificity 60.5%). AdaBoost had higher sensitivity but at the cost of extremely poor specificity - meaning it flagged too many non-cancerous regions as suspicious.

The Dice ratios (measuring spatial overlap of detected cancer with ground truth) were 62.0% in the peripheral zone, 56.7% in the central gland, and notably 78.0% in the transition zone - the hardest area to detect. This is particularly significant because transition zone cancer is most easily confused with benign prostatic hyperplasia, yet the algorithm performed best there.

The combination of all three MRI types (T2, DWI, and dADC) gave the best SBE of 87.1%, superior to any pair of modalities. Among single modalities, dADC alone (83.2%) outperformed DWI alone (80.4%) and T2 alone (76.8%), confirming that diffusion-based imaging carries the most discriminatory information for prostate cancer detection.

TL;DR: The algorithm achieved 87.1% section-based accuracy and 91.5% sensitivity, substantially outperforming four comparison methods and performing well across all three prostate zones including the difficult transition zone.
Pages 13-15
Clinical Applications: Biopsy Guidance, Focal Therapy, and Active Surveillance

The most direct clinical application is MRI-guided prostate biopsy. By accurately localizing which sections contain cancer before the biopsy is performed, the system could concentrate needle placements on genuinely suspicious regions rather than using a random sampling strategy. This could reduce the number of biopsy cores needed, lower complication risks (bleeding, infection), and improve cancer detection rates - particularly for tumors missed by standard TRUS-guided systematic biopsy.

For patients diagnosed with low-risk prostate cancer who choose active surveillance (monitoring rather than immediate treatment), the system could provide consistent, repeatable MRI-based tracking of tumor location and extent over time. Standardized automated tumor localization would remove variability between serial MRI reads and objectively detect changes in cancer burden that might signal a need to transition from surveillance to treatment.

In the setting of focal therapy - minimally invasive treatments targeting only the cancer-containing areas (such as high-intensity focused ultrasound, cryotherapy, or laser ablation) - precise cancer localization directly translates to better treatment planning. Knowing exactly which sections contain cancer allows ablation to be concentrated on the tumor while preserving surrounding healthy tissue, including nerves critical for sexual and urinary function.

The framework is also potentially extendable to other cancers. The study demonstrates this by applying the same algorithm to brain tumor localization using a 20-patient brain MRI dataset, achieving a mean Dice ratio of 0.768. This adaptability suggests the core approach of iterative random forests with auto-context could serve as a general platform for multi-organ, multi-cancer detection from multi-parametric imaging.

TL;DR: Automated prostate cancer localization enables MRI-guided biopsy, consistent active surveillance monitoring, precise focal therapy targeting, and potentially extends to detection of other cancers.
Page 15
Conclusions: Toward Fully Automated, Whole-Prostate Cancer Detection

This study demonstrates a novel fully automatic framework for localizing prostate cancer in the entire prostate using multiparametric MRI. By combining random forests with an iterative auto-context model that incorporates previously estimated cancer probability maps as spatial context features, the method overcomes the limitations of prior semi-automatic or peripheral-zone-only approaches.

With 87.1% section-based accuracy, 91.5% sensitivity, 85.1% specificity, and AUC values of 0.832 (voxel) and 0.883 (section), the system substantially outperforms established classification methods including SVM and AdaBoost. It achieves strong performance across all three prostate zones including the diagnostically challenging transition zone, without requiring any manual region outlining.

Key limitations include the small dataset (26 patients) and the reliance on cancer delineations from only 1-3 MRI slices per patient, which may cause low-grade tumors with subtle imaging features to be missed. Future work will focus on validation in larger patient cohorts and may incorporate additional MRI sequences to further improve accuracy - moving toward a clinically deployable tool that could transform prostate cancer detection, biopsy guidance, and treatment planning.

TL;DR: A fully automated random forest plus auto-context framework achieves 87.1% accuracy for whole-prostate cancer localization on MRI, outperforming all comparison methods and enabling practical biopsy guidance and treatment planning.
Citation: Open Access, . Available at: PMC5508576.