Add model card for COMPOSE

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +46 -0
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: text-generation
4
+ tags:
5
+ - mathematics
6
+ ---
7
+
8
+ # COMPOSE: Composing Future Theorems from Citations and Formal Structure
9
+
10
+ COMPOSE is a dual-graph framework that generates predicted future mathematical claims for a given anchor paper. It leverages two complementary sources of context: the scientific citation graph and an aligned formal theorem dependency graph from Mathlib.
11
+
12
+ - **Paper:** [COMPOSE: Composing Future Theorems from Citations and Formal Structure](https://huggingface.co/papers/2605.30333)
13
+ - **Project Page:** [https://david-busbib.github.io/COMPOSE-page/](https://david-busbib.github.io/COMPOSE-page/)
14
+ - **Repository:** [https://github.com/david-busbib/COMPOSE](https://github.com/david-busbib/COMPOSE)
15
+
16
+ ## Model Description
17
+
18
+ The COMPOSE architecture jointly encodes a paper's citation subgraph and its formal Mathlib4 theorem dependencies.
19
+
20
+ - **Encoder:** A dual-graph GNN encoder fuses these signals via bidirectional cross-attention.
21
+ - **Decoder:** A DeepSeek-Math-7B decoder is conditioned on the fused graph representations to produce predicted mathematical claims in plain text.
22
+
23
+ The model is designed to produce claims that are both grounded in prior work and motivated by formal mathematical structures.
24
+
25
+ ## Usage
26
+
27
+ To use COMPOSE, follow the installation instructions in the [official GitHub repository](https://github.com/david-busbib/COMPOSE). You can run the model on a specific arXiv paper using the provided script:
28
+
29
+ ```bash
30
+ python3 code/run_on_paper.py --arxiv_id <ARXIV_ID> --checkpoint path/to/compose_checkpoint.pt
31
+ ```
32
+
33
+ ## Results
34
+
35
+ Experiments show that COMPOSE outperforms strong baselines on retrieval tasks for real future papers and achieves superior performance under LLM-judge evaluation, producing more grounded and mathematically richer outputs.
36
+
37
+ ## Citation
38
+
39
+ ```bibtex
40
+ @article{busbib2026compose,
41
+ title={{COMPOSE}: Composing Future Theorems from Citations and Formal Structure},
42
+ author={Busbib, David and Werman, Michael},
43
+ journal={arXiv preprint arXiv:2605.30333},
44
+ year={2026}
45
+ }
46
+ ```