nielsr HF Staff commited on
Commit
1734d07
·
verified ·
1 Parent(s): a60a695

Improve model card and add metadata

Browse files

This PR improves the model card for the POST model.

Summary of changes:
- Added `pipeline_tag: other` to the metadata.
- Included links to the paper and official GitHub repository.
- Added key highlights from the paper.
- Included sample usage (evaluation scripts) directly from the GitHub README.
- Updated the citation section with the provided BibTeX.

Files changed (1) hide show
  1. README.md +38 -1
README.md CHANGED
@@ -1,10 +1,47 @@
1
  ---
2
  license: apache-2.0
 
3
  ---
4
 
5
- Pretrained POST models, please consider citing our work:
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  ```
 
 
 
 
 
 
8
  @misc{zhang2026postpriorobservationadversariallearning,
9
  title={POST: Prior-Observation Adversarial Learning of Spatio-Temporal Associations for Multivariate Time Series Anomaly Detection},
10
  author={Suofei Zhang and Yaxuan Zheng and Haifeng Hu},
 
1
  ---
2
  license: apache-2.0
3
+ pipeline_tag: other
4
  ---
5
 
6
+ # POST: Prior-Observation Adversarial Learning of Spatio-Temporal Associations for Multivariate Time Series Anomaly Detection
7
 
8
+ This repository contains the official PyTorch implementation and pre-trained checkpoints for the paper [POST: Prior-Observation Adversarial Learning of Spatio-Temporal Associations for Multivariate Time Series Anomaly Detection](https://huggingface.co/papers/2605.18128).
9
+
10
+ ## 💡 Highlights
11
+ - **Novel Framework**: We propose POST, a model that unifies spatial and temporal anomaly modeling under a joint prior–observation adversarial learning framework.
12
+ - **Anomaly Localization**: The adversarial optimization not only improves time-wise detection but also enables the model to localize anomalies to specific channels.
13
+ - **SMD+ Dataset**: Construction of a dataset featuring precise channel-wise annotations, serving as a generic testbed for anomaly localization in MTS.
14
+ - **SOTA Performance**: Extensive experiments demonstrate that POST consistently outperforms state-of-the-art (SOTA) baselines in standard anomaly detection tasks.
15
+
16
+ ## 🛠️ Resources
17
+ - **Code**: [GitHub Repository](https://github.com/anocodetest1/POST)
18
+ - **Paper**: [huggingface.co/papers/2605.18128](https://huggingface.co/papers/2605.18128)
19
+
20
+ ## 🚀 Get Started
21
+
22
+ ### Evaluation
23
+ To evaluate the trained models on the benchmarks, run the following scripts. Ensure you have the datasets prepared and the pre-trained weights placed into the `./checkpoints/` folder (named as `<dataset_name>_checkpoint.pth`).
24
+
25
+ ```bash
26
+ # SMD
27
+ python main.py --anomaly_ratio 0.5 --batch_size 64 --mode test --dataset SMD --data_path dataset/SMD --input_c 38 --output_c 38
28
+ # MSL
29
+ python main.py --anomaly_ratio 1. --batch_size 64 --mode test --dataset MSL --data_path dataset/MSL --input_c 55 --output_c 55
30
+ # SMAP
31
+ python main.py --anomaly_ratio 1. --batch_size 64 --mode test --dataset SMAP --data_path dataset/SMAP --input_c 25 --output_c 25
32
+ # SWaT
33
+ python main.py --anomaly_ratio 1. --win_size 150 --batch_size 64 --mode test --dataset SWaT --data_path dataset/SWaT --input_c 51 --output_c 51
34
+ # PSM
35
+ python main.py --anomaly_ratio 1. --batch_size 64 --mode test --dataset PSM --data_path dataset/PSM --input_c 25 --output_c 25
36
+ # SMD+
37
+ python main_channel.py --anomaly_ratio 0.5 --batch_size 64 --mode test --dataset SMD+ --data_path dataset/SMD+ --input_c 38 --output_c 38
38
  ```
39
+
40
+ ## Suggested citation
41
+
42
+ Please consider citing our work:
43
+
44
+ ```bibtex
45
  @misc{zhang2026postpriorobservationadversariallearning,
46
  title={POST: Prior-Observation Adversarial Learning of Spatio-Temporal Associations for Multivariate Time Series Anomaly Detection},
47
  author={Suofei Zhang and Yaxuan Zheng and Haifeng Hu},