In this paper, a neural architecture search (NAS) framework is proposed for 3D medical image segmentation, to automatically optimize a neural architecture from a large design space. Our NAS framework searches the structure of each layer including neural connectivities and operation types in both of the encoder and decoder. Since optimizing over a large discrete architecture space is difficult due to high-resolution 3D medical images, a novel stochastic sampling algorithm based on a continuous relaxation is also proposed for scalable gradient based optimization. On the 3D medical image segmentation tasks with a benchmark dataset, an automatically designed architecture by the proposed NAS framework outperforms the human-designed 3D U-Net, and moreover this optimized architecture is well suited to be transferred for different tasks.
Background
An angiography‐based supervised machine learning (
ML
) algorithm was developed to classify lesions as having fractional flow reserve ≤0.80 versus >0.80.
Methods and Results
With a 4:1 ratio, 1501 patients with 1501 intermediate lesions were randomized into training versus test sets. Between the ostium and 10 mm distal to the target lesion, a series of angiographic lumen diameter measurements along the centerline was plotted. The 24 computed angiographic features based on the diameter plot and 4 clinical features (age, sex, body surface area, and involve segment) were used for
ML
by XGBoost. The model was independently trained and tested by 2000 bootstrap iterations. External validation with 79 patients was conducted. Including all 28 features, the
ML
model with 5‐fold cross‐validation in the 1204 training samples predicted fractional flow reserve ≤0.80 with overall diagnostic accuracy of 78±4% (averaged area under the curve: 0.84±0.03). The 12 high‐ranking features selected by scatter search were involved segment; body surface area; distal lumen diameter; minimal lumen diameter; length of a lumen diameter <2.0 mm, <1.5 mm, and <1.25 mm; mean lumen diameter within the worst segment; sex; diameter stenosis; distal 5‐mm reference lumen diameter; and length of diameter stenosis >70%. Using those 12 features, the
ML
predicted fractional flow reserve ≤0.80 in the test set with sensitivity of 84%, specificity of 80%, and overall accuracy of 82% (area under the curve: 0.87). The averaged diagnostic accuracy in bootstrap replicates was 81±1% (averaged area under the curve: 0.87±0.01). External validation showed accuracy of 85% (area under the curve: 0.87).
Conclusions
Angiography‐based
ML
showed good diagnostic performance in identifying ischemia‐producing lesions and reduced the need for pressure wires.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.