s
stella4528/DeepPlant-GEP-npu
模型介绍
文件和版本
Pull Requests
讨论
分析

DeepPlant-GEP — 植物基因表达预测(昇腾 NPU 适配)

DeepPlant 是序列级植物调控基因组学基础模型。本仓库为 GEP (Gene Expression Prediction) 子任务模型,输入固定 2500 bp DNA 序列(one-hot 编码),输出 2935 个基因表达特征(expm1 逆变换后)。 约 248.67M 参数,F32,MIT 许可。

架构:ConvNet backbone(反向互补卷积 + 残差卷积塔)→ 6 层 TransformerEncoder(embed 2048 / 8 头)→ AttentionPool → GEP 预测头。

已在 Ascend 910(CANN 8.5.1 / torch_npu 2.9.0.post1)上完成真实推理 验证:NPU_INFERENCE_PASS=True,finite 检查通过,权重 strict 加载。

  • 代码:https://github.com/Addaoud/DeepPlant
  • 论文:Deep-Plant: a supervised foundation model for plant regulatory genomics(bioRxiv 2026)

推理脚本 inference.py,部署说明见 readme.md,真实 NPU 验证证据见 results/。