Unsupervised Learning Composite Network to Reduce Training Cost of Deep Learning Model for Colorectal Cancer Diagnosis.

Journal: IEEE journal of translational engineering in health and medicine
Published Date:

Abstract

Deep learning facilitates complex medical data analysis and is increasingly being explored in colorectal cancer diagnostics. However, the training cost of the deep learning model limits its real-world medical utility. In this study, we present a composite network that combines deep learning and unsupervised K-means clustering algorithm (RK-net) for automatic processing of medical images. RK-net was more efficient in image refinement compared with manual screening and annotation. The training of a deep learning model for colorectal cancer diagnosis was accelerated by two times with utilization of RK-net-processed images. Better performance was observed in training loss and accuracy achievement as well. RK-net could be useful to refine medical images of the ever-expanding quantity and assist in subsequent construction of the artificial intelligence model.

Authors

  • Jirui Guo
    Department of Colorectal SurgeryThe Sixth Affiliated Hospital, Sun Yat-sen University Guangzhou 510655 China.
  • Wuteng Cao
    Department of Colorectal Surgery, Guangdong Institute of Gastroenterology, Guangdong Provincial Key Laboratory of Colorectal and Pelvic Floor Diseases, Department of Radiology, The Sixth Affiliated Hospital of Sun Yat-Sen University, Guangzhou, Guangdong, China.
  • Bairun Nie
    School of Electrical Computer and Telecommunications EngineeringUniversity of Wollongong Wollongong NSW 2522 Australia.
  • Qiyuan Qin
    Department of Colorectal SurgeryThe Sixth Affiliated Hospital, Sun Yat-sen University Guangzhou 510655 China.