marvinsxtr commited on
Commit
8d4bb4c
·
0 Parent(s):

Upload models

Browse files
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: icmil
4
+ tags:
5
+ - multiple-instance-learning
6
+ - in-context-learning
7
+ - prior-data-fitted-network
8
+ - tabular
9
+ datasets:
10
+ - bifold-pathomics/ICMIL
11
+ pipeline_tag: tabular-classification
12
+ ---
13
+
14
+ # ICMIL — In-Context Multiple Instance Learning
15
+
16
+ 📄 [Paper (arXiv:2606.06458)](https://arxiv.org/abs/2606.06458) · 💻 [Code](https://github.com/injurise/ICMIL) · 🤗 [Datasets](https://huggingface.co/datasets/bifold-pathomics/ICMIL)
17
+
18
+ ICMIL is an in-context learner for bag-structured data: pretrained on synthetic MIL
19
+ tasks, it labels the bags of a new task from a handful of labelled bags in a single
20
+ forward pass, with no gradient updates, fine-tuning or hyper-parameter search.
21
+
22
+ This repository holds the three trained model seeds behind the paper's
23
+ **ICMIL (Ours)** results. The same checkpoints are committed in the code repository
24
+ under [`checkpoints/`](https://github.com/injurise/ICMIL/tree/main/checkpoints).
25
+
26
+ ## Files
27
+
28
+ | File | Seed name |
29
+ |---|---|
30
+ | `icmil-c5trd795.pt` | `c5trd795` |
31
+ | `icmil-ggwsqibd.pt` | `ggwsqibd` |
32
+ | `icmil-k337zhz1.pt` | `k337zhz1` |
33
+
34
+ Seed names are opaque run labels. Each `.pt` holds
35
+ `{model_state_dict, epoch, ...provenance}`. The reported ICMIL row is the
36
+ **mean ± cross-seed SEM** over the three seeds.
37
+
38
+ We also provide three 50-feature seeds —
39
+ `icmil-50dim-ldsebntm.pt`, `icmil-50dim-sjik4hzb.pt`, `icmil-50dim-zeqy97nm.pt`.
40
+ No paper results are based on them.
41
+
42
+ ## Usage
43
+
44
+ ```python
45
+ from icmil import load_icmil # pip install -e . from https://github.com/injurise/ICMIL
46
+
47
+ model = load_icmil(seed="c5trd795", device="cuda")
48
+ # X_train: (1, n_ctx_bags, bag_size, n_features), y_train: (1, n_ctx_bags)
49
+ # X_test: (1, n_query_bags, bag_size, n_features)
50
+ logits = model(X_train, y_train, X_test) # (1, n_query_bags, n_classes)
51
+ ```
52
+
53
+ `load_icmil` reads `checkpoints/` in the repo by default; point it elsewhere with
54
+ `load_icmil(source="/path/to/ckpts", seed=...)` or `ICMIL_CKPT_DIR`. Reproduce the
55
+ full benchmark table with `python -m icmil.reproduce`.
56
+
57
+ ## Citation
58
+
59
+ ```bibtex
60
+ @article{mollers2026incontext,
61
+ title = {In-Context Multiple Instance Learning},
62
+ author = {M\"ollers, Alexander and Sextro, Marvin and Hense, Julius and Dernbach, Gabriel and M\"uller, Klaus-Robert},
63
+ journal = {arXiv preprint arXiv:2606.06458},
64
+ year = {2026}
65
+ }
66
+ ```
icmil-50dim-ldsebntm.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be5e285e30d6a7202f71169540bdca3d25ac22610d09d5265b1ec7a868ab32cb
3
+ size 99319989
icmil-50dim-sjik4hzb.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30115cb5f942bb79450c9149d1b09aa2962bd98963a3f0ff7307f70da53ade27
3
+ size 99319989
icmil-50dim-zeqy97nm.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0602ad321bedcee7d6772c9eaf8274059c8f326f7a1f4a04b1948e7c3a16bcee
3
+ size 99319989
icmil-c5trd795.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0fb477e8bbc108e73880d10eabcb6b60835f36fd1b7ec2b59af432d43a4322aa
3
+ size 99250764
icmil-ggwsqibd.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5588caf2f7f15bcd2cbd309d1c5663fdf2f95200dc1f4e23faed4a2a4507f520
3
+ size 99250764
icmil-k337zhz1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0829de6a1a28c212fcb284a2240c9b4bb866601ce3c0b0b9e9cd5076b4e931c
3
+ size 99250764