If recent research is any indication, artificial intelligence (AI) has a bright future in medicine. Nvidia developed an AI system that can generate synthetic scans of brain cancer. Google subsidiary DeepMind has demonstrated a machine learning algorithm that can recommend treatment for more than 50 eye diseases with 94% accuracy. And in newly published research, New York University (NYU) showed how AI might aid in lung cancer diagnosis.
A paper published in the journal Nature Medicine (“Classification and mutation prediction from non-small cell lung cancer histopathology images using deep learning”) describes how a team of NYU researchers retrained Google’s Inception v3, an open source convolutional neural network architected for object identification, to detect certain forms of lung cancers with 97% accuracy.
That’s encouraging news for the more than 200,000 people diagnosed with lung cancer each year. More than 150,000 people die annually as a result of disease-related complications, according to the American Cancer Society and the Cancer Statistics Center.
The team trained, validated, and tested Inception v3 on 1,634 whole-slide images from The Cancer Genome Atlas (TCGA), a public dataset maintained by the National Cancer Institute (NCI) and National Human Genome Research Institute (NHGRI) that maps genomic changes in 33 types of cancer. Once it began reliably—with 99% accuracy—identifying cancerous cells, the team taught it to tell the difference between two of the most common forms of lung cancer: adenocarcinoma (LUAD) and squamous cell carcinoma (LUSC).
It performed well when let loose on independent samples taken from cancer patients at NYU, correctly diagnosing images between 83% and 97% of the time. That’s despite encountering features it hadn’t previously seen, including blood clots, blood vessels, inflammation, necrotic regions, and regions of collapsed lung.
All the more impressively, the model took just 20 seconds on average, running on a single-GPU PC, to calculate classification probabilities.
Next, the researchers trained it to identify not just cancerous tissue, but the genetic mutations within the tissue. Fed images from TCGA and genetic profiles for each tumor, the system drew on subtle differences in the tumors’ appearance to predict six of the ten most commonly mutated genes in LUAD (STK11, EGFR, FAT1, SETBP1, KRAS, and TP53).
In the future, the team hopes to extend the model’s classification to other, less common lung cancers such as large-cell carcinoma, in addition to “histological subtypes” and features such as necrosis, fibrosis, and others. According to Wired, they’re considering forming a company and seeking Federal Food and Drug Administration (FDA) approval to commercialize their technology.