CIR-Net: Automatic Classification of Human Chromosome based on Inception-ResNet Architecture

Abstract

Method: This paper focuses on chromosome classification because it is critical for chromosome karyotyping. In recent years, deep learning-based methods are the most promising methods for solving the tasks of chromosome classification. Although the deep learning-based Inception architecture has yielded state-of-the-art performance in the 2015 ILSVRC challenge, it has not been used in chromosome classification tasks so far. Therefore, we develop an automatic chromosome classification approach named CIR-Net based on Inception-ResNet which is an optimized version of Inception. However, the classification performance of origin Inception-ResNet on the insufficient chromosome dataset still has a lot of capacity for improvement. Further, we propose a simple but effective augmentation method called CDA for improving the performance of CIR-Net. Results: The experimental results show that our proposed method achieves 95.98% classification accuracy on the clinical G-band chromosome dataset whose training dataset is insufficient. Moreover, the proposed augmentation method CDA improves more than 8.5% (from 87.46% to 95.9%) classification accuracy compared to other methods. In this paper, the experimental results demonstrate that our proposed method is recent the most effective solution for solving clinical chromosome classification problems in chromosome auto-karyotyping on the condition of the insufficient training dataset. Code and Dataset are available at https://github.com/CloudDataLab/CIR-Net.