Researchers from the group led by Julia Vogt, who runs the Medical Data Science Group at ETH Zurich, recently teamed up with neonatologists at KUNO Klinik St. Hedwig to develop a computer model that provides reliable support in diagnosing the disease in newborn infants. Their results have now been published in the International Journal of Computer Vision.
Making AI reliable and explainable
The ETH researchers began by training their algorithm on hundreds of video recordings taken from ultrasound examinations of the hearts of 192 newborns. This dataset also included moving images of the beating heart taken from different angles as well as diagnoses by experienced paediatric cardiologists (is pulmonary hypertension present or not) and an evaluation of the disease’s severity (“mild” or “moderate to severe”). To determine the algorithm’s success at interpreting the images, the researchers subsequently added a second dataset of ultrasound images from 78 newborn infants, which the model had never seen before. The model suggested the correct diagnosis in around 80 to 90 percent of cases and was able to determine the correct level of disease severity in around 65 to 85 percent of cases.
“The key to using a machine-learning model in a medical context is not just the prediction accuracy, but also whether humans are able to understand the criteria the model uses to make decisions,” Vogt says. Her model makes this possible by highlighting the parts of the ultrasound image on which its categorisation is based. This allows doctors to see exactly which areas or characteristics of the heart and its blood vessels the model considered to be suspicious. When the paediatric cardiologists examined the datasets, they discovered that the model looks at the same characteristics as they do, even though it was not explicitly programmed to do so.
A human makes the diagnosis
This machine-learning model could potentially be extended to other organs and diseases, for example to diagnose heart septal defects or valvular heart disease.
It could also be useful in regions where no specialists are available: standardised ultrasound images could be taken by a healthcare professional, and the model could then provide a preliminary risk assessment and an indication of whether a specialist should be consulted. Medical facilities that do have access to highly qualified specialists could use the model to ease their workload and to help reach a better and more objective diagnosis. “AI has the potential to make significant improvements to healthcare. The crucial issue for us is that the final decision should always be made by a human, by a doctor. AI should simply be providing support to ensure that the maximum number of people can receive the best possible medical care,” Vogt says.
