Publications & Preprints

@article{kong2024plmix,
	title={On Unsupervised Domain Adaptation: Pseudo Label Guided Mixup for Adversarial Prompt Tuning},
	author={Fanshuang Kong, Richong Zhang, Ziqiao Wang and Yongyi Mao},
	booktitle={Proceedings of the AAAI conference on artificial intelligence},
	year={2024},
}

@inproceedings{huang2024cusa,
  title={Cross-modal and Uni-modal Soft-label Alignment for Image-Text Retrieval},
  author={Huang, Hailang and Nie, Zhijie and Wang, Ziqiao and Shang, Ziyu},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  year={2024},
}

@inproceedings{
Wang2023sampleconditioned,
title={Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Bounds},
author={Ziqiao Wang and Yongyi Mao},
booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
year={2023},
url={https://openreview.net/forum?id=oqDSDKLd3S}
}

@inproceedings{
    wang2022on,
    title={On the Generalization of Models Trained with {SGD}: Information-Theoretic Bounds and Implications},
    author={Ziqiao Wang and Yongyi Mao},
    booktitle={International Conference on Learning Representations},
    year={2022},
    url={https://openreview.net/forum?id=oWZsQ8o5EA}
}

@inproceedings{Wang2022a,
  author = {Ziqiao Wang and Yongyi Mao},
  title = {Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal States},
  booktitle = {Proceedings of the Fortieth Conference on Uncertainty in Artificial Intelligence},
  year = {2024},
  url={https://openreview.net/pdf?id=ZhMkLTWRyh}
}

@inproceedings{
	wang2023two,
	title={Two Facets of {SDE} Under an Information-Theoretic Lens: Generalization of {SGD} via Training Trajectories and via Terminal States},
	author={Ziqiao Wang and Yongyi Mao},
	booktitle={NeurIPS 2023 Workshop on Mathematics of Modern Machine Learning},
	year={2023},
	url={https://openreview.net/forum?id=hWDqKtIwSo}
}

@inproceedings{
    liu2023overtraining,
    title={Over-Training with Mixup May Hurt Generalization},
    author={Zixuan Liu and Ziqiao Wang and Hongyu Guo and Yongyi Mao},
    booktitle={The Eleventh International Conference on Learning Representations },
    year={2023},
    url={https://openreview.net/forum?id=JmkjrlVE-DG}
}

@article{Wang2020,
  author = {Ziqiao Wang and Yongyi Mao and Hongyu Guo and Richong Zhang},
  title = {On SkipGram Word Embedding Models with Negative Sampling: Unified Framework and Impact of Noise Distributions},
  journal = {CoRR},
  year = {2020},
  volume = {abs/2009.04413},
  url = {https://arxiv.org/abs/2009.04413}
}

@inproceedings{
    wang2023informationtheoretic,
    title={Information-Theoretic Analysis of Unsupervised Domain Adaptation},
    author={Ziqiao Wang and Yongyi Mao},
    booktitle={The Eleventh International Conference on Learning Representations },
    year={2023},
    url={https://openreview.net/forum?id=c5tbxWXU9-y}
}

@inproceedings{wang2023tighter,
  title={Tighter Information-Theoretic Generalization Bounds from Supersamples},
  author={Wang, Ziqiao and Mao, Yongyi},
  booktitle={International Conference on Machine Learning},
  year={2023},
  organization={PMLR}
}

* denotes equal contribution.

Preprints

Conference Publications

  1. Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal States
    Ziqiao Wang and Yongyi Mao
    Conference on Uncertainty in Artificial Intelligence (UAI) 2024, [Arxiv][BibTeX].

  2. On Unsupervised Domain Adaptation: Pseudo Label Guided Mixup for Adversarial Prompt Tuning
    Fanshuang Kong, Richong Zhang, Ziqiao Wang and Yongyi Mao
    Proceedings of the AAAI Conference on Artificial Intelligence (AAAI) 2024, [Code][Poster][BibTeX].

  3. Cross-modal and Uni-modal Soft-label Alignment for Image-Text Retrieval
    Hailang Huang, Zhijie Nie, Ziqiao Wang and Ziyu Shang
    Proceedings of the AAAI Conference on Artificial Intelligence (AAAI) 2024, [Arxiv][Code][Poster][BibTeX].

  4. Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Bounds
    Ziqiao Wang and Yongyi Mao
    Advances in Neural Information Processing Systems (NeurIPS) 2023, [Arxiv] [Poster] [BibTeX].

  5. Tighter Information-Theoretic Generalization Bounds from Supersamples
    Ziqiao Wang and Yongyi Mao
    International Conference on Machine Learning (ICML) 2023 (Oral Presentation, top 2.2% of submissions), [Arxiv][Code][Poster][BibTeX].

  6. Information-Theoretic Analysis of Unsupervised Domain Adaptation
    Ziqiao Wang and Yongyi Mao
    International Conference on Learning Representations (ICLR) 2023, [Arxiv][Poster][BibTeX].

  7. Over-Training with Mixup May Hurt Generalization
    Zixuan Liu*, Ziqiao Wang*, Hongyu Guo, and Yongyi Mao
    International Conference on Learning Representations (ICLR) 2023, [Arxiv][BibTeX].

  8. On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implications
    Ziqiao Wang and Yongyi Mao
    International Conference on Learning Representations (ICLR) 2022, [Arxiv][Poster][BibTeX].

Workshop Publications

  1. On f-Divergence Principled Domain Adaptation: An Improved Framework
    Ziqiao Wang and Yongyi Mao
    18th Canadian Workshop on Information Theory (CWIT 2024).

  2. Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal States
    Ziqiao Wang and Yongyi Mao
    NeurIPS 2023 Workshop on Mathematics of Modern Machine Learning (M3L), [Poster] [BibTeX].

  3. Over-Training with Mixup May Hurt Generalization
    Zixuan Liu*, Ziqiao Wang*, Hongyu Guo, and Yongyi Mao
    NeurIPS 2022 Workshop on Interpolation Regularizers and Beyond, [Poster].