arXiv · 1808.08277
Deep Mask For X-ray Based Heart Disease Classification
Abstract
We build a deep learning model to detect and classify heart disease using $X-ray$. We collect data from several hospitals and public datasets. After preprocess we get 3026 images including disease type VSD, ASD, TOF and normal control. The main problem we have to solve is to enable the network to accurately learn the characteristics of the heart, to ensure the reliability of the network while increasing accuracy. By learning the doctor's diagnostic experience, labeling the image and using tools to extract masks of heart region, we train a U-net to generate a mask to give more attention. It forces the model to focus on the characteristics of the heart region and obtain more reliable results.
Explore related subjects
Keep this discovery
Xupeng Chen, Binbin Shi. 2018-08-19. Deep Mask For X-ray Based Heart Disease Classification. https://arxiv.org/abs/1808.08277
Cite the original work for its findings. Save a collection to share your selection of sources.