hawthorneluke commited on
Commit
4b20df3
·
verified ·
1 Parent(s): 6a995f5

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +24 -0
  2. model.pt +3 -0
README.md ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags: [astronomy, supernova, regression, ztf, timm, convnext]
3
+ datasets: [MultimodalUniverse/btsbot]
4
+ license: mit
5
+ ---
6
+ # Supernova Peak Predictor
7
+ Predicts **when** and **how bright** a supernova will become from early ZTF observations.
8
+
9
+ ## Novel Contribution
10
+ First model to predict `days_to_peak` and `peakmag` as regression targets from rise-phase alerts.
11
+
12
+ ## Architecture
13
+ ConvNeXt-pico (Galaxy Zoo pretrained) + 23-feature metadata MLP -> fusion -> dual regression heads
14
+ 8,811,816 params (8,425,576 trainable)
15
+
16
+ ## Test Results
17
+ | Metric | days_to_peak | peakmag |
18
+ |--------|-------------|---------|
19
+ | MAE | 124.03 days | 0.324 mag |
20
+ | Median AE | 29.15 days | 0.223 mag |
21
+
22
+ ## Data
23
+ [MultimodalUniverse/btsbot](https://huggingface.co/datasets/MultimodalUniverse/btsbot) — rise-phase supernovae only
24
+ 3x63x63 image triplets + 23 metadata features. Leaky features (age, days_since_peak) excluded.
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa4877b500635de02378ce122fd6d2e104bf3b0979a720bdfbf98297be42145f
3
+ size 35284727