A Robust Framework of Chromosome Straightening with ViT-Patch GAN

Sifan Song,Jinfeng Wang,Fengrui Cheng,Qirui Cao,Yihan Zuo,Yongteng Lei,Ruomai Yang,Chunxiao Yang,Frans Coenen,Jia Meng,Kang Dang,Jionglong Su
DOI: https://doi.org/10.48550/arXiv.2203.02901
2022-03-06
Computer Vision and Pattern Recognition
Abstract:Chromosomes exhibit non-rigid and non-articulated nature with varying degrees of curvature. Chromosome straightening is an essential step for subsequent karyotype construction, pathological diagnosis and cytogenetic map development. However, robust chromosome straightening remains challenging, due to the unavailability of training images, distorted chromosome details and shapes after straightening, as well as poor generalization capability. We propose a novel architecture, ViT-Patch GAN, consisting of a motion transformation generator and a Vision Transformer-based patch (ViT-Patch) discriminator. The generator learns the motion representation of chromosomes for straightening. With the help of the ViT-Patch discriminator, the straightened chromosomes retain more shape and banding pattern details. The proposed framework is trained on a small dataset and is able to straighten chromosome images with state-of-the-art performance for two large datasets.
What problem does this paper attempt to address?