Hejie Cui

Hejie Cui

Postdoctoral Scholar

Stanford University

Biography

Hi! I am a Postdoctoral Scholar working with Professor Nigam Shah on AI for Healthcare at Stanford University. I am also fortunate to receive mentorship from Professor Sanmi Koyejo on Trustworthy ML. I received my Ph.D. in Computer Science at Emory University, advised by Professor Carl Yang and worked closely with Professor Joyce C Ho on ML for Health. In the past summers, I interned at Microsoft Research and Amazon Science.

Research Interests: My research interests focuses on Large Language Model (LLM) Evaluation and Post-training for Healthcare. I’m happy to mentor a few students on research and especially encourage students from underrepresented groups to reach out. If you are a junior student and want to discuss research and/or look for advice, feel free to schedule a chat via this link.

Previously, I got my bachelor’s degree (Valedictorian, GPA Ranking: 1/164) in Computer Science and Technology at Tongji University. During my undergraduate, I worked on computer vision for medical imaging with Prof. Lin Zhang. In addition, I worked with Prof. Tianwei Yu on machine learning for bioinformatics. I also spent a wonderful summer working with Prof. Gabor Fichtinger at Queen’s University in Canada through the Mitacs program.

News

  • [2024.09] Our work on clinician preference aligned synthetic instruction generation for visual instruction tuning is accepted to NeurIPS'24 Datasets and Benchmarks Track.
  • [2024.09] Our work on network recall by large language models is accepted to NeurIPS'24 Research Track. Abstract version is presented on IC2S2'24 as an Oral.
  • [2024.08] Excited to be selected as a Rising Star Spotlight Speaker to give a talk about my research at University of Michigan 2024 AI Symposium.
  • [2024.06] Our work on multi-agent LLM reasoning for EHR-based few-shot disease prediction is accepted to AMIA Annual Symposium as an Oral.
  • [2024.05] Our work on disease subtyping is accepted to KDD Applied Data Science Track.
  • [2024.04] My PhD Work is selected to the CHIL Doctoral Symposium. Thank you CHIL!
  • [2024.04] I successfully defended my dissertation. Officially, Dr. Cui!
  • [2024.03] Our survey paper on LLM domain specialization is cited by the 2024 Economic Report of the President!
  • [2023.12] Glad to received NSF Student Travel Support Award for the ICDM 2023.
  • [2023.09] Our paper on artificial node features on non-attributed graphs has been selected as Most Influential CIKM Papers produced by Best Paper Digest.
  • [2023.09] One work on visual knowledge extraction is accepted to NeurIPS'23.
  • [2023.09] Two work are accepted to PSB'24. Congrats to Alexis (High-Schooler)!
  • [2023.08] Humbled to be selected as a 2023 EECS Rising Star!
  • [2023.05] One work on biological data augmentation is accepted to KDD'23.
  • [2023.05] One work on multimodal extraction is accepted to ACL'23 Findings.
  • [2023.04] Our work on brain network pre-training is accepted to CHIL'23 as an Oral.
  • [2022.11] Our paper on few-shot learning is accepted to AAAI'23 as an Oral.
  • [2022.11] Glad to receive NeurIPS AI4Science Travel Award!
  • [2022.10] Our Benchmark paper on Graph Neural Networks for brain networks has now been officially accepted to IEEE TMI.
  • [2022.09] Our paper on brain transformer is accepted to NeurIPS'22 as an Spotlight.
  • [2022.08] One paper on node feature for non-attributed graphs is accepted to CIKM'22.
  • [2022.06] One paper on interpretable GNNs is accepted to MICCAI'22 as an Oral.

Interests

  • AI for Healthcare
  • Large (Vision) Language Models
  • Data Mining and Data Science

Education

  • Ph.D. in Computer Science, 2019-now

    Emory University

  • B.Eng. in Software Engineering, 2015-2019

    Tongji University

Experience

 
 
 
 
 

Research Intern

Microsoft Research

May 2023 – Aug 2023 Redmond, Washington, United States
Worked on graph recommendation in the joint supervision of augmented learning and reasoning team, office of applied AI research, and MSAI.
 
 
 
 
 

Applied Scientist Intern

Amazon

May 2022 – Aug 2022 Seattle, Washington, United States
Worked on multimodality knowledge extraction in the product knowledge graph team.
 
 
 
 
 

Mitacs Global Research Intern

Queens University

Jul 2018 – Oct 2018 Kingston, Ontario, Canada
Visited Perk Lab and worked on video detection for surgery training with Prof. Gabor Fichtinger.
 
 
 
 
 

Microsoft University Summer Camp

Microsoft Research Asia

Aug 2017 – Aug 2017 Beijing, China
our project on prototyping workflow for HoloLens mix-reality got the best work award of Microsoft student summer camp hackathon.
 
 
 
 
 

SAP Student Training and Rotation Program

SAP

Jun 2017 – Aug 2017 Shanghai, China
Worked on ABAP database.

Selected Accomplishments

  • Rising Stars in EECS, 2023
  • KDD Student Travel Grant Award, 2023
  • NeurIPS AI4Science Travel Award, 2022
  • NSF Student Travel Grant Award for CIKM, 2022
  • Laney Graduate Student Council Research Grant, Emory University, 2022
  • MICCAI Student Travel Grant Award, 2022
  • Award for CRA-WP Grad Cohort for Women, 2021
  • Mitacs Globalink Research Award, 2018
  • Valedictorian of the School of Software Engineering Class 2019, Tongji University, 2019
  • Outstanding Graduate Award, the City of Shanghai, 2019
  • Outstanding Graduate Award, Tongji University, 2019
  • Tongji University Programming Competition, Silver Prize, 2017
  • National Scholarship, 2016, 2017, 2018

Talks

  • Advancing Healthcare with Multimodal Structural Knowledge
    University of California, Los Angeles. Oct 2023
    Stanford University. Nov 2023
    Massachusetts General Hospital, Harvard Medical School. Nov 2023
    Zhejiang University. Dec 2023
    Shanghai Jiao Tong University, Fudan University, Tongji University. Dec 2023
    Peking University. Jan 2024
  • BrainGB: A Benchmark for Brain Network Analysis with Graph Neural Networks
    University of Rochester. Apr 2021
  • Effective and Interpretable Graph Neural Networks for Graphs Constructed from Multimodality Data
    Microsoft Search, Assistant and Intelligence (MSAI). Apr 2021

Services

  • Program Committee Member / Reviewer:
    • Conference: ACL'24, ICML'24, ICLR'24; NeurIPS'23; ML4H'23; CHIL'24,23; SIGKDD'23,22; ICDM'22; WWW'22; AAAI'23,22,21; SDM'22; LOG'22; MICCAI'23,22,20;
    • Journal: TKDE, TMI, IEEE BigData, Health Data Science
    • Workshop: Interpretable Machine Learning in Healthcare ICML'22,21; Computer Vision for Automated Medical Diagnosis ICCV'21; Medical Imaging meets NeurIPS'21; Temporal Graph Learning NeurIPS'22; Graph-Based Natural Language Processing COLING'22
  • Organizer:
    • Neural Networks for Brain Connectome Analysis: Theories, Methods, and Applications Workshop IEEE BigData'22
    • Federated Learning with Graph Data Workshop CIKM'22
  • Volunteer:
    • Conference: CHIL'23; CIKM'22; ICML'21,20
  • Teaching:
    • CS 253 Data Structures and Algorithms, TA (Spring 2021)
    • CS 584 Natural Language Processing for Biomedical Applications, Head TA (Fall 2020)
    • CS 584 Deep Learning, Head TA (Spring 2020)
  • Leadership: Chair, Microsoft Student Club, Tongji University (2017-2018)