avatar

Yiwei Wang

Postdoc
UCLA
yiweiwang2@ucmerced.edu


About Me

I will join the Department of Computer Science at University of California, Merced as an Assistant Professor in the Spring of 2025.

Recruiting Ph.D. students and research interns. I’m actively looking for strong and motivated students. If you are interested in working with me, please feel free to email me.

I am currently a Postdoc at UCLA advised by Prof. Kai-Wei Chang and Prof. Nanyun Peng. I received my Ph.D. in Computer Science at National University of Singapore, where I was fortunate to be advised by Prof. Bryan Hooi and collaborate with Prof. Muhao Chen. My recent research is focused on trustworthy large language models.

Selected Publications [Google Scholar][DBLP]

    2024

  1. Fast Graph Sharpness-Aware Minimization for Enhancing and Accelerating Few-Shot Node Classification
    Yihong Luo, Yuhan Chen, Siya Qiu, Yiwei Wang, Chen Zhang, Yan Zhou, Xiaochun Cao, Jing Tang.
    Advances in Neural Information Processing Systems (NeurIPS), 2024.
    PDF Code
  2. Adaptive Token Biaser: Knowledge Editing via Biasing Key Entities
    Baolong Bi, Shenghua Liu, Yiwei Wang, Lingrui Mei, Hongcheng Gao, Yilong Xu, Xueqi Cheng.
    Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    PDF Code
  3. Control Large Language Models via Divide-and-Conquer
    Bingxuan Li, Yiwei Wang, Tao Meng, Nanyun Peng, Kai-Wei Chang.
    Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    PDF Code
  4. SLANG: New Concept Comprehension of Large Language Models
    Lingrui Mei, Shenghua Liu, Yiwei Wang, Baolong Bi, Xueqi Chen.
    Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    PDF Code
  5. AlignedCoT: Prompting Large Language Models via Native-Speaking Demonstrations
    Zhicheng Yang, Yinya Huang, Jing Xiong, Liang Feng, Xiaodan Liang, Yiwei Wang, Jing Tang.
    Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    PDF Code
  6. LLM-A* : Large Language Model Enhanced Incremental Heuristic Search on Path Planning
    Silin Meng, Yiwei Wang, Cheng-Fu Yang, Nanyun Peng, Kai-Wei Chang.
    Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    PDF Code
  7. Lpnl: Scalable link prediction with large language models
    Baolong Bi, Shenghua Liu, Yiwei Wang, Lingrui Mei, Xueqi Cheng.
    Association for Computational Linguistics ACL 2024, 2024.
    PDF Code
  8. Scalable and Effective Implicit Graph Neural Networks on Large Graphs
    Juncheng Liu, Bryan Hooi, Kenji Kawaguchi, Yiwei Wang, Chaosheng Dong, Xiaokui Xiao.
    The Twelfth International Conference on Learning Representations (ICLR), 2024.
    PDF Code
  9. Decoding by Contrasting Knowledge: Enhancing LLMs' Confidence on Edited Facts
    Baolong Bi, Shenghua Liu, Lingrui Mei, Yiwei Wang, Pengliang Ji, Xueqi Cheng.
    arXiv preprint arXiv:2405.11613, 2024.
    PDF Code
  10. Is factuality decoding a free lunch for llms? evaluation on knowledge editing benchmark
    Baolong Bi, Shenghua Liu, Yiwei Wang, Lingrui Mei, Xueqi Cheng.
    arXiv preprint arXiv:2404.00216, 2024.
    PDF Code
  11. UniTST: Effectively Modeling Inter-Series and Intra-Series Dependencies for Multivariate Time Series Forecasting
    Juncheng Liu, Chenghao Liu, Gerald Woo, Yiwei Wang, Bryan Hooi, Caiming Xiong, Doyen Sahoo.
    arXiv preprint arXiv:2406.04975, 2024.
    PDF Code
  12. "Not Aligned" is Not" Malicious": Being Careful about Hallucinations of Large Language Models' Jailbreak
    Lingrui Mei, Shenghua Liu, Yiwei Wang, Baolong Bi, Jiayi Mao, Xueqi Cheng.
    arXiv preprint arXiv:2406.11668, 2024.
    PDF Code
  13. Preprint
    Deepedit: Knowledge editing as decoding with constraints
    Yiwei Wang, Muhao Chen, Nanyun Peng, Kai-Wei Chang.
    arXiv preprint arXiv:2401.10471, 2024.
    PDF Code
  14. Frustratingly Easy Jailbreak of Large Language Models via Output Prefix Attacks
    Yiwei Wang, Muhao Chen, Nanyun Peng, Kai-Wei Chang.
    Preprint, 2024.
    PDF Code

    15. 2023

  15. A Causal View of Entity Bias in (Large) Language Models
    Fei Wang, Wenjie Mo, Yiwei Wang, Wenxuan Zhou, Muhao Chen.
    Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2023.
    PDF Code
  16. Primacy Effect of ChatGPT
    Yiwei Wang, Yujun Cai, Muhao Chen, Yuxuan Liang, Bryan Hooi.
    Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2023.
    PDF Code
  17. Airformer: Predicting nationwide air quality in china with transformers
    Yuxuan Liang, Yutong Xia, Songyu Ke, Yiwei Wang, Qingsong Wen, Junbo Zhang, Yu Zheng, Roger Zimmermann.
    Proceedings of the AAAI Conference on Artificial Intelligence, 2023.
    PDF Code
  18. How Fragile is Relation Extraction under Entity Replacements?
    Yiwei Wang, Bryan Hooi, Fei Wang, Yujun Cai, Yuxuan Liang, Wenxuan Zhou, Jing Tang, Manjuan Duan, Muhao Chen.
    CONLL 2023, 2023.
    PDF Code
  19. Graph explicit neural networks: Explicitly encoding graphs for efficient and accurate inference
    Yiwei Wang, Bryan Hooi, Yozen Liu, Neil Shah.
    Proceedings of the Sixteenth ACM International Conference on Web Search and Data Mining (WSDM), 2023.
    PDF Code

    20. 2022

  20. Dangling-Aware Entity Alignment with Mixed High-Order Proximities
    Juncheng Liu, Zequn Sun, Bryan Hooi, Yiwei Wang, Dayiheng Liu, Baosong Yang, Xiaokui Xiao, Muhao Chen.
    Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics (NAACL), 2022.
    PDF Code
  21. GraphCache: Message Passing as Caching for Sentence-Level Relation Extraction
    Yiwei Wang, Muhao Chen, Wenxuan Zhou, Yujun Cai, Yuxuan Liang, Bryan Hooi.
    Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics (NAACL), 2022.
    PDF Code
  22. Should We Rely on Entity Mentions for Relation Extraction? Debiasing Relation Extraction with Counterfactual Analysis
    Yiwei Wang, Muhao Chen, Wenxuan Zhou, Yujun Cai, Yuxuan Liang, Dayiheng Liu, Baosong Yang, Juncheng Liu, Bryan Hooi.
    Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics (NAACL), 2022.
    PDF Code
  23. Mixed-order relation-aware recurrent neural networks for spatio-temporal forecasting
    Yuxuan Liang, Kun Ouyang, Yiwei Wang, Zheyi Pan, Yifang Yin, Hongyang Chen, Junbo Zhang, Yu Zheng, David S Rosenblum, Roger Zimmermann.
    IEEE Transactions on Knowledge and Data Engineering (TKDE), 2022.
    PDF Code
  24. Lscale: latent space clustering-based active learning for node classification
    Juncheng Liu, Yiwei Wang, Bryan Hooi, Renchi Yang, Xiaokui Xiao.
    Joint European Conference on Machine Learning and Knowledge Discovery in Databases (ECML-PKDD), 2022.
    PDF Code
  25. Flashlight: Scalable link prediction with effective decoders
    Yiwei Wang, Bryan Hooi, Yozen Liu, Tong Zhao, Zhichun Guo, Neil Shah.
    Learning on Graphs Conference (LOG), 2022.
    PDF Code

    26. 2021

  26. Eignn: Efficient infinite-depth graph neural networks
    Juncheng Liu, Kenji Kawaguchi, Bryan Hooi, Yiwei Wang, Xiaokui Xiao.
    Advances in Neural Information Processing Systems (NeurIPS), 2021.
    PDF Code
  27. Adaptive data augmentation on temporal graphs
    Yiwei Wang, Yujun Cai, Yuxuan Liang, Henghui Ding, Changhu Wang, Siddharth Bhatia, Bryan Hooi.
    Advances in Neural Information Processing Systems (NeurIPS), 2021.
    PDF Code
  28. A unified 3d human motion synthesis model via conditional variational auto-encoder
    Yujun Cai, Yiwei Wang, Yiheng Zhu, Tat-Jen Cham, Jianfei Cai, Junsong Yuan, Jun Liu, Chuanxia Zheng, Sijie Yan, Henghui Ding, others.
    Proceedings of the IEEE/CVF International Conference on Computer Vision, 2021.
    PDF Code
  29. Structure-aware label smoothing for graph neural networks
    Yiwei Wang, Yujun Cai, Yuxuan Liang, Wei Wang, Henghui Ding, Muhao Chen, Jing Tang, Bryan Hooi.
    arXiv preprint arXiv:2112.00499, 2021.
    PDF Code
  30. Time-aware neighbor sampling for temporal graph networks
    Yiwei Wang, Yujun Cai, Yuxuan Liang, Henghui Ding, Changhu Wang, Bryan Hooi.
    2022 International Joint Conference on Neural Networks (IJCNN), 2021.
    PDF Code
  31. Fine-grained urban flow prediction
    Yuxuan Liang, Kun Ouyang, Junkai Sun, Yiwei Wang, Junbo Zhang, Yu Zheng, David Rosenblum, Roger Zimmermann.
    Proceedings of the Web Conference 2021, 2021.
    PDF Code
  32. Modeling Trajectories with Neural Ordinary Differential Equations.
    Yuxuan Liang, Kun Ouyang, Hanshu Yan, Yiwei Wang, Zekun Tong, Roger Zimmermann.
    IJCAI, 2021.
    PDF Code
  33. Revisiting convolutional neural networks for citywide crowd flow analytics
    Yuxuan Liang, Kun Ouyang, Yiwei Wang, Ye Liu, Junbo Zhang, Yu Zheng, David S Rosenblum.
    Machine Learning and Knowledge Discovery in Databases: European Conference, (ECML-PKDD), 2021.
    PDF Code
  34. Curgraph: Curriculum learning for graph classification
    Yiwei Wang, Wei Wang, Yuxuan Liang, Yujun Cai, Bryan Hooi.
    Proceedings of the Web Conference 2021 (WWW), 2021.
    PDF Code
  35. Mixup for node and graph classification
    Yiwei Wang, Wei Wang, Yuxuan Liang, Yujun Cai, Bryan Hooi.
    Proceedings of the Web Conference 2021 (WWW), 2021.
    PDF Code
  36. Progressive supervision for node classification
    Yiwei Wang, Wei Wang, Yuxuan Liang, Yujun Cai, Bryan Hooi.
    Machine Learning and Knowledge Discovery in Databases: European Conference, (ECML-PKDD), 2021.
    PDF Code

    37. 2020

  37. Active learning for node classification: The additional learning ability from unlabelled nodes
    Juncheng Liu, Yiwei Wang, Bryan Hooi, Renchi Yang, Xiaokui Xiao.
    arXiv preprint arXiv:2012.07065, 2020.
    PDF Code
  38. Graphcrop: Subgraph cropping for graph classification
    Yiwei Wang, Wei Wang, Yuxuan Liang, Yujun Cai, Bryan Hooi.
    arXiv preprint arXiv:2009.10564, 2020.
    PDF Code
  39. Learning progressive joint propagation for human motion prediction
    Yujun Cai, Lin Huang, Yiwei Wang, Tat-Jen Cham, Jianfei Cai, Junsong Yuan, Jun Liu, Xu Yang, Yiheng Zhu, Xiaohui Shen, others.
    Computer Vision--ECCV 2020: 16th European Conference, 2020.
    PDF Code
  40. Revisiting convolutional neural networks for urban flow analytics
    Yuxuan Liang, Kun Ouyang, Yiwei Wang, David Samuel Rosenblum.
    Preprint, 2020.
    PDF Code
  41. Detecting implementation bugs in graph convolutional network based node classifiers
    Yiwei Wang, Wei Wang, Yujun Ca, Bryan Hooi, Beng Chin Ooi.
    2020 IEEE 31st International Symposium on Software Reliability Engineering (ISSRE), 2020.
    PDF Code
  42. KDD
    NodeAug: Semi-supervised node classification with data augmentation
    Yiwei Wang, Wei Wang, Yuxuan Liang, Yujun Cai, Juncheng Liu, Bryan Hooi.
    Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, 2020.
    PDF Code
  43. Provably robust node classification via low-pass message passing
    Yiwei Wang, Shenghua Liu, Minji Yoon, Hemank Lamba, Wei Wang, Christos Faloutsos, Bryan Hooi.
    2020 IEEE International Conference on Data Mining (ICDM), 2020.
    PDF Code

    44. 2019

  44. Optimization algorithms for graph Laplacian estimation via ADMM and MM
    Licheng Zhao, Yiwei Wang, Sandeep Kumar, Daniel P Palomar.
    IEEE Transactions on Signal Processing, 2019.
    PDF Code

    45. 2017

  45. Using knowledge graphs to explain entity co-occurrence in Twitter
    Yiwei Wang, Mark James Carman, Yuan-Fang Li.
    Proceedings of the 2017 ACM on Conference on Information and Knowledge Management, 2017.
    PDF Code

Teaching

Services

Conference Reviewers

Journal Reviewers

Contact

Address: 405 Hilgard Avenue, LA 90095, CA
Office Location: UCLA Eng VI Rm 397


Last Update: