跳到正文
Briefings in Bioinformatics·· 3 小时前AI 评分52

HiCP2GAN:以 Hi-C 基础模型为判别器的即插即用 GAN 增强框架

HiCP2GAN: a plug and play foundation model-based GAN for Hi-C enhancement

AI 导读

HiCP2GAN 是一个即插即用的 GAN 框架,用预训练的 Vision Transformer 类 Hi-C 基础模型充当判别器,以提升低分辨率 Hi-C 接触图谱的质量。该判别器在跨物种、跨细胞类型的 1.18 亿个 Hi-C patch 上预训练,为对抗训练提供具有生物学意义的梯度;框架与生成器无关,任何兼容的 Hi-C 分辨率增强架构都可作为生成器,便于方法间公平比较。作者将基础模型的编码器阶段改作判别器骨干,微调靠近输入的浅层、冻结更深的 transformer 层,在保留预训练知识的同时做任务适配。

阅读原文

来源:Briefings in Bioinformatics · doi.org

论文摘要

The 3D organization of chromatin shapes gene regulation and cellular function, and Hi-C has become the primary technique for mapping chromatin interactions genome-wide. However, high-resolution data remain costly and scarce, leaving many studies with sparse contact maps that limit downstream analysis. Deep learning methods, especially generative adversarial networks (GANs), have shown promise for enhancing low-resolution Hi-C data, but most existing GAN-based approaches rely on custom discriminators trained from scratch, which can yield unstable training and limited generalization. Hi-C foundation models pretrained on large-scale data capture rich, transferable representations of chromatin structure, yet their use as discriminators within adversarial enhancement frameworks had not been explored. We introduce HiCP2GAN, a plug-and-play GAN that employs a pretrained Vision Transformer-based Hi-C foundation model as its discriminator. The discriminator was pretrained on 118 million Hi-C patches across diverse species and cell types, providing biologically meaningful gradients for adversarial supervision. The framework is generator-agnostic: any compatible Hi-C resolution enhancement architecture can serve as the generator, enabling fair comparison across methods. The encoder phase of the foundation model was adapted as a discriminator backbone, finetuning the first few layers from the input while freezing the deeper transformer layers to preserve pretrained knowledge while allowing task-specific adaptation. Experiments across various cell lines show that HiCP2GAN outperforms traditional GAN-based methods in most settings and improves the reconstruction quality and biological reproducibility of existing Hi-C enhancement models in most evaluated cases. HiCP2GAN is publicly available at https://github.com/OluwadareLab/HiCP2GAN.