Xiaoliang Shi commited on
Commit
1805de1
·
verified ·
1 Parent(s): 4604cb5

Add model card

Browse files
Files changed (1) hide show
  1. README.md +82 -0
README.md ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ arxiv: 2607.20057
3
+ license: mit
4
+ tags:
5
+ - protein
6
+ - antibody
7
+ - antibody-design
8
+ - multimodal
9
+ - biological-sequence-modeling
10
+ - structure-aware
11
+ ---
12
+
13
+ # AAMFM: Antigen-specific Antibody Multi-modal Foundation Model
14
+
15
+ **AAMFM** is a multi-modal foundation model for antigen-specific functional
16
+ antibody design. It jointly models antibody sequence and structure, conditioned
17
+ on antigen geometry and epitope annotations.
18
+
19
+ [[Paper]](https://arxiv.org/abs/2607.20057)
20
+ [[Code]](https://github.com/XL-S224/AAMFM)
21
+
22
+ <p align="center">
23
+ <img src="https://raw.githubusercontent.com/XL-S224/AAMFM/main/assets/aamfm-overview.png" alt="Overview of AAMFM" width="100%">
24
+ </p>
25
+ <p align="center"><em>Hierarchical training and inference pipeline of AAMFM.</em></p>
26
+
27
+ ## Model description
28
+
29
+ AAMFM follows a hierarchical training strategy:
30
+
31
+ 1. **Antibody-domain adaptation:** continual pre-training on paired antibody
32
+ sequence--structure data.
33
+ 2. **Antigen-specific supervised fine-tuning:** a lightweight adapter fuses
34
+ antigen geometric features with epitope information.
35
+ 3. **Preference alignment:** calibrated DPO aligns the model with
36
+ structure-based functional preferences and sequence plausibility.
37
+
38
+ At inference time, the model is conditioned on an antigen, epitope-site
39
+ annotations, and antibody framework sequences to generate CDR sequences and a
40
+ complete antibody structure.
41
+
42
+ ## Checkpoint
43
+
44
+ | File | Description |
45
+ | --- | --- |
46
+ | **AAMFM.pt** | AAMFM model checkpoint. |
47
+
48
+ ## Usage
49
+
50
+ Use this checkpoint with the official
51
+ [AAMFM code repository](https://github.com/XL-S224/AAMFM). The repository
52
+ contains the training, generation, evaluation, and pseudo-log-likelihood
53
+ utilities, together with instructions for the required external assets.
54
+
55
+ Generation requires a compatible base-model backend, its externally supplied
56
+ weights, antigen geometric features, and antigen/epitope inputs. See the
57
+ [repository README](https://github.com/XL-S224/AAMFM#readme) for the complete
58
+ environment and runtime setup.
59
+
60
+ ## Intended use and limitations
61
+
62
+ This checkpoint is intended for research in computational antibody design. Its
63
+ outputs are computational candidates, not experimentally validated antibodies,
64
+ and must not be used for clinical, diagnostic, or therapeutic decisions without
65
+ appropriate experimental validation and expert review.
66
+
67
+ ## Citation
68
+
69
+ If you use AAMFM, please cite:
70
+
71
+ @article{shi2026aamfm,
72
+ title={Antigen-specific Antibody Multi-modal Foundation Model for Functional Antibody Design},
73
+ author={Shi, Xiaoliang and Wang, Zichen and Ma, Runze and Zhang, Zhongyue and Zheng, Shuangjia},
74
+ journal={arXiv preprint arXiv:2607.20057},
75
+ year={2026}
76
+ }
77
+
78
+ ## License
79
+
80
+ The AAMFM checkpoint and the accompanying
81
+ [source-code repository](https://github.com/XL-S224/AAMFM) are released under
82
+ the [MIT License](https://opensource.org/license/mit).