VITA

Repositories

• TransGAN

  [Preprint] "TransGAN: Two Transformers Can Make One Strong GAN", Yifan Jiang, Shiyu Chang, Zhangyang Wang

  pytorch transformer gan transformer-encoder transformer-models
  Python 24 447 3 0 Updated Feb 18, 2021

• Alleviate-Robust-Overfitting

  [ICLR 2021] "Robust Overfitting may be mitigated by properly learned smoothening" by Tianlong Chen*, Zhenyu Zhang*, Sijia Liu, Shiyu Chang, Zhangyang Wang

  smoothing overfitting adversarial-robustness generalization-ability
  Python MIT 1 4 0 0 Updated Feb 18, 2021

• VITA-Group.github.io
  HTML 0 0 0 0 Updated Feb 16, 2021

• Unified-LTH-GNN

  [Preprint] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhangyang Wang

  pruning co-design graph-neural-networks graph-sparsification lottery-ticket-hypothesis co-optimization
  Python MIT 0 3 0 0 Updated Feb 15, 2021

• GAN-LTH

  [ICLR 2021] "GANs Can Play Lottery Too" by Xuxi Chen, Zhenyu Zhang, Yongduo Sui, Tianlong Chen

  transfer generative-adversarial-network gan pruning lottery-ticket-hypothesis
  Python MIT 1 4 0 0 Updated Feb 9, 2021

• Nasty-Teacher

  [ICLR 2021 Spotlight Oral] "Undistillable: Making A Nasty Teacher That CANNOT teach students", Haoyu Ma, Tianlong Chen, Ting-Kuei Hu, Chenyu You, Xiaohui Xie, Zhangyang Wang

  copyright knowledge-distillation teacher-student privacy-protection undistillable
  Python MIT 1 3 0 0 Updated Feb 5, 2021

• Privacy-AdversarialLearning

  [ECCV 2018] Towards Privacy-Preserving Visual Recognition via Adversarial Training: A Pilot Study
  Jupyter Notebook 9 25 1 0 Updated Feb 5, 2021

• FasterSeg

  [ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang

  latency pytorch semantic-segmentation cityscapes tensorrt neural-architecture-search
  Python MIT 89 426 4 1 Updated Feb 5, 2021

• TENAS

  [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, Zhangyang Wang

  neural-architecture-search ntk nas-bench-201 darts-space linear-region
  Python MIT 5 30 0 0 Updated Feb 2, 2021

• KD-ticket
  Python MIT 1 2 0 0 Updated Feb 1, 2021

• EnlightenGAN

  [IEEE TIP'2021] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang

  pytorch gan unsupervised-learning generative-adversarial-networks low-light-enhance
  Python 128 424 7 0 Updated Feb 1, 2021

• Once-for-All-Adversarial-Training

  [NeurIPS 2020] "Once-for-All Adversarial Training: In-Situ Tradeoff between Robustness and Accuracy for Free" by Haotao Wang*, Tianlong Chen*, Shupeng Gui, Ting-Kuei Hu, Ji Liu, and Zhangyang Wang
  Python MIT 4 20 0 0 Updated Jan 18, 2021

• L2O-Minimax

  [ICLR 2021] "Learning a Minimax Optimizer: A Pilot Study" by Jiayi Shen*, Xiaohan Chen*, Howard Heaton*, Tianlong Chen, Jialin Liu, Wotao Yin, Zhangyang Wang
  Python MIT 0 2 0 0 Updated Jan 17, 2021

• Lifelong-Learning-LTH

  [ICLR 2021] "Long Live the Lottery: The Existence of Winning Tickets in Lifelong Learning" by Tianlong Chen*, Zhenyu Zhang*, Sijia Liu, Shiyu Chang, Zhangyang Wang

  lth lifelong-learning lottery-ticket-hypothesis continue-learning winning-tickets
  Python MIT 0 2 0 0 Updated Jan 15, 2021

• PA-HMDB51

  [TPAMI 2020] "Privacy-Preserving Deep Action Recognition: An Adversarial Learning Framework and A New Dataset" by Zhenyu Wu, Haotao Wang, Zhaowen Wang, Hailin Jin, and Zhangyang Wang

  dataset privacy-protection human-action-recognition privacy-attributes
  Python 4 26 0 0 Updated Jan 3, 2021

• SmartDeal

  [Preprint] "SmartDeal: Re-Modeling Deep Network Weights for Efficient Inference and Training", Xiaohan Chen, Yang Zhao, Yue Wang, Pengfei Xu, Haoran You, Chaojian Li, Yonggan Fu, Yingyan Lin, and Zhangyang Wang
  MIT 0 0 0 0 Updated Dec 31, 2020

• L2O-Training-Techniques

  [NeurIPS 2020 Spotlight Oral] "Training Stronger Baselines for Learning to Optimize", Tianlong Chen*, Weiyi Zhang*, Jingyang Zhou, Shiyu Chang, Sijia Liu, Lisa Amini, Zhangyang Wang

  imitation-learning meta-learning curriculum-learning learning-to-learn training-tricks self-improving learning-to-optimize
  Python MIT 5 6 0 0 Updated Dec 27, 2020

• L2-GCN

  [CVPR 2020] L2-GCN: Layer-Wise and Learned Efficient Training of Graph Convolutional Networks

  paper graph-convolutional-networks gcn learning-to-learn efficient-training
  Python 11 19 0 0 Updated Dec 27, 2020

• CV_LTH_Pre-training

  [Preprint] "The Lottery Tickets Hypothesis for Supervised and Self-supervised Pre-training in Computer Vision Models" Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu, Yang Zhang, Michael Carbin, Zhangyang Wang

  transfer transfer-learning moco pre-training lottery-ticket-hypothesis simclr simclrv2
  0 20 0 0 Updated Dec 14, 2020

• AutoGAN

  [ICCV 2019] "AutoGAN: Neural Architecture Search for Generative Adversarial Networks" by Xinyu Gong, Shiyu Chang, Yifan Jiang and Zhangyang Wang

  pytorch gan cifar10 neural-architecture-search autogan
  Python MIT 70 376 1 0 Updated Dec 11, 2020

• Self-PU

  [ICML2020] "Self-PU: Self Boosted and Calibrated Positive-Unlabeled Training" by Xuxi Chen, Wuyang Chen, Tianlong Chen, Ye Yuan, Chen Gong, Kewei Chen, Zhangyang Wang

  pytorch pu-learning
  Python MIT 8 24 0 0 Updated Dec 8, 2020

• GAN-Slimming

  [ECCV 2020] "All-in-One GAN Compression by Unified Optimization" by Haotao Wang, Shupeng Gui, Haichuan Yang, Ji Liu, and Zhangyang Wang

  compression pytorch gan cyclegan
  Python MIT 16 72 3 0 Updated Nov 30, 2020

• 3D_Adversarial_Logo

  [Preprint] "Can 3D Adversarial Logos Cloak Humans?"

  cloak detection detector logos 3d-mesh adversarial-attacks
  Python MIT 2 11 0 0 Updated Nov 27, 2020

• Adversarial-Contrastive-Learning

  [NeurIPS 2020] “ Robust Pre-Training by Adversarial Contrastive Learning”, Ziyu Jiang, Tianlong Chen, Ting Chen, Zhangyang Wang
  Python 6 33 0 0 Updated Oct 29, 2020

• GraphCL

  Forked from Shen-Lab/GraphCL

  [NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen
  Python MIT 17 0 0 0 Updated Oct 27, 2020

• BERT-Tickets

  [NeurIPS 2020] "The Lottery Ticket Hypothesis for Pre-trained BERT Networks", Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu, Yang Zhang, Zhangyang Wang, Michael Carbin

  bert lottery-tickets pre-training lottery-ticket-hypothesis universal-embeddings
  Python MIT 6 85 2 0 Updated Oct 27, 2020

• AGD

  [ICML2020] "AutoGAN-Distiller: Searching to Compress Generative Adversarial Networks" by Yonggan Fu, Wuyang Chen, Haotao Wang, Haoran Li, Yingyan Lin, Zhangyang Wang

  compression pytorch generative-adversarial-network neural-architecture-search
  Python MIT 14 78 1 0 Updated Oct 25, 2020

• FracTrain

  Forked from RICE-EIC/FracTrain

  [NeurIPS'2020] "FracTrain: Fractionally Squeezing Bit Savings Both Temporally and Spatially for Efficient DNN Training", Yonggan Fu, Haoran You, Yang Zhao, Yue Wang, Chaojian Li, Kailash Gopalakrishnan, Zhangyang Wang, and Yingyan Lin
  Python 3 0 0 0 Updated Oct 25, 2020

• mm-hand

  [ACM'MM 2020] "MM-Hand: 3D-Aware Multi-Modal Guided Hand Generative Network for 3D Hand Pose Synthesis" Zhenyu Wu, Duc Hoang, Shih-Yao Lin, Yusheng Xie, Liangjian Chen, Yen-Yu Lin, Zhangyang Wang, Wei Fan
  Python 1 9 0 0 Updated Oct 7, 2020

• ShiftAddNet

  Forked from RICE-EIC/ShiftAddNet

  [NeurIPS 2020] "ShiftAddNet: A Hardware-Inspired Deep Network", Haoran You, Xiaohan Chen, Yongan Zhang, Chaojian Li, Sicheng Li, Zihao Liu, Zhangyang Wang, Yingyan Lin
  Python MIT 7 0 0 0 Updated Oct 4, 2020