Implementation of the AlignReg package associated with the following paper:

Haohan Wang, Zeyi Huang, Xindi Wu, and Eric P. Xing. 2022. Toward Learning Robust and Invariant Representations with Alignment Regular- ization and Data Augmentation. In Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD ’22), August 14–18, 2022, Washington, DC, USA. ACM, New York, NY, USA,

The package is organized by Hanru Yan

One can direclty install the package with

  pip install alignreg 

Or one can download the code and use it locally.

One can use the package in simply one line of code.

Please visit the iPython example for usage instructions in both TensorFlow and PyTorch.

This repository serves for the purpose to guide others to use our tool, if you are interested in the scripts to replicate our results in paper, please contact us and we will share the repository for replication.

Haohan Wang · Hanru Yan