Home

Awesome

treatment-causal-transformer

Treatment Learning Causal Transformer for Noisy Image Classification

Treatment Learning Transformer Code

python causal_trans.py

Details in line (427) of the class Causal_Transformer

Treatment Effect Code

Average Treatment Effect (ATE) batch evaluation

eval.py

Baseline setup:

Task: Image Binary Classification (w/wo dogs)
X: COCO Dataset image w/wo dogs (NICO benchmark) or w/wo noise (CPS)
Z: Dog segments
T: 1 if has car else 0 with some noise
Y: 1 if has car else 0
<img src="https://github.com/huckiyang/treatment-causal-transformer/blob/main/tlt.png" width="900">

If you find this work is related to your research, please consider to cite the following work. Thank you!

@inproceedings{yang2023treatment,
  title={Treatment Learning Causal Transformer for Noisy Image Classification},
  author={Yang, Chao-Han Huck and Hung, I-Te and Liu, Yi-Chieh and Chen, Pin-Yu},
  booktitle={Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision},
  pages={6139--6150},
  year={2023}
}