About Me

Introduction

I am a postdoctoral associate at Cornell University, under the supervision of Prof. Fengqi You. Before that, I was a postdoctoral research associate at the University of Illinois Urbana-Champaign (UIUC), working with Prof. Shaowen Wang and Prof. Jiawei Han. In August 2025, I received my PhD in Computer Science from Florida International University (FIU), advised by Prof. Giri Narasimhan. In 2024 Summer, I was a Momentum Research Fellow at Columbia University.

With the goal of AI for environmental sustainability, my research focuses on LLM, Agent and AI for Envionemental and Climate Science. I've been pursuing my research goals from data mining (structured retrieval and reasoning: HyperCube-RAG, MultiCube-RAG, TaSR-RAG), model development (Agentic AI, LLM, diffusion model: PyRAG, CoDiCast, ReFine), and scientific applications (flood hazard prediction and risk mitigation WaLeF, FIDLAr, Water-RAF). I have multiple publications at AAAI (Oral), KDD, ICML, IJCAI, EMNLP, IEEE BigData, Journal of Water Resources Planning and Management, Journal of Hydraulic Engineering, and Nature Machine Intelligence.

Sincere gratitude to funding sources: NSF I-GUIDE led by UIUC, NSF LEAP led by Columbia University.

[Job Hunting] I am on the job market for AI research positions in both academia and industry. If you believe I might be a good fit, please feel free to reach out at jimengshi008@gmail.com.

Highlights

• [Talk] Presented at the Florida House Natural Resources & Disasters Subcommittee.
• [Talk] Invited talk at the U.S. National Oceanic and Atmospheric Administration (NOAA).
• [Talk] Invited talk about "AI for Flood Prediction and Mitigation" at Columbia University.
• [Talk] Invited talk about "Toward Envionemental Intelligence" at Peking University (Online).
• [Talk] Guest lecturer @UIUC CS598JH: Fall 2025 “Text Mining with LLMs”.
• [Media] Reported/featured by Florida Internationa University (FIU) News.
• [Media] Reported/featured by The Year in Science: The most impactful research stories of 2025.
• [Media] Reported/featured by South Florida Business Journal.
• [Media] Reported/featured by The Chronicle of Higher Education.
• [Award] Momentum Research Fellowship from Columbia University.
• [Award] 2025 Geospatial Rising Star (50 awardees/year worldwide).
• [Award] Best Paper Award, World Environmental and Water Resources Congress 2023.

Selected Publications [Full List]

• MultiCube-RAG for Multi-hop Question Answering
  Jimeng Shi, Wei Hu, Runchu Tian, Bowen Jin, Wonbin Kweon, SeongKu Kang, Yunfan Kang, Dingqi Ye, Sizhe Zhou, Shaowen Wang, Jiawei Han
  (Under review)
  
  
• Hypercube-RAG: Hypercube Retrieval-Augmented Generation for Scientific Question-Answering
  Jimeng Shi, Sizhe Zhou, Bowen Jin, Wei Hu, Runchu Tian, Shaowen Wang, Giri Narasimhan, Jiawei Han
  (Under review)
  
  
• Retrieval is Cheap, Show Me the Code: Executable Multi-Hop Reasoning for Retrieval-Augmented Generation
  Jiashuo Sun^*, Jimeng Shi^*, Yixuan Xie, Saizhuo Wang, Jash Pajesh Parekh, Pengcheng Jiang, Zhiyi Shi, Jiajun Fan, Qinglong Zheng, Peiran Li, Shaowen Wang, Ge Liu, Jiawei Han
  (Under review)
  
  
• Deep Learning and Foundation Models for Weather Prediction: A Survey
  Jimeng Shi, Azam Shirali, Bowen Jin, Sizhe Zhou, Wei Hu, Rahuul Rangaraj, Zhaonan Wang, Yanzhao Wu, Leonardo Bobadilla, Upmanu Lall, Shaowen Wang, Jiawei Han, Giri Narasimhan
  The 35th International Joint Conference on Artificial Intelligence (IJCAI 2026)
  
  
• Retrieval-Augmented Foundation Models for Water Level Prediction in the Everglades
  Rahuul Rangaraj^*,†, Jimeng Shi^*,#, Rajendra Paudel, Giri Narasimhan^#, Yanzhao Wu^#
  The 32nd SIGKDD Conference on Knowledge Discovery and Data Mining (KDD 2026)
  
  
• FIDLAR: Forecast-Informed Deep Learning Approaches for Flood Mitigation
  Jimeng Shi, Zeda Yin, Arturo Leon, Jayantha Obeysekera, Giri Narasimhan
  The 39th Annual AAAI Conference on Artificial Intelligence (AAAI 2025) (Oral)
  
  
• CoDiCast: Conditional Diffusion Model for Weather Prediction with Uncertainty Quantification
  Jimeng Shi, Bowen Jin, Jiawei Han, Sundararaman Gopalakrishnan, Giri Narasimhan
  The 34th International Joint Conference on Artificial Intelligence (IJCAI 2025) (Oral)
  
  
• Deep Learning Models for Water Level Prediction in South Florida
  Jimeng Shi, Zeda Yin, Rukmangadh Myana, Arturo Leon, Jayantha Obeysekera, Giri Narasimhan
  Journal of Water Resources Planning and Management, 151(10)
  
  
• ReFine: Boosting Time Series Prediction Under Extreme Events by Reweighting and Fine-tuning
  Jimeng Shi, Azam Shirali, Giri Narasimhan
  2024 IEEE International Conference on Big Data (IEEE BigData 2024) (Oral)
  
  
• The Power of Explainability in Forecast-Informed Deep Learning Models for Flood Mitigation
  Jimeng Shi, Vitalii Stebliankin, Giri Narasimhan
  NeurIPS 2023 workshop on Climate Change AI
  
  
• Graph Transformer Network for Flood Forecasting with Heterogeneous Covariates
  Jimeng Shi, Vitalii Stebliankin, Zhaonan Wang, Shaowen Wang, Giri Narasimhan
  NSF I-GUIDE Forum in New York, 2023 (Oral)
  
  
• TimeX++: Learning Time-Series Explanations with Information Bottleneck
  Zichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng, Zhuomin Chen, Lei Song, Wenqian Dong, Jayantha Obeysekera, Farhad Shirani, Dongsheng Luo
  The 41st International Conference on Machine Learning (ICML 2024)
  
  
• Evaluating protein binding interfaces with transformer networks
  Vitalii Stebliankin, Azam Shirali, Prabin Baral, Jimeng Shi, Prem Chapagain, Kalai Mathee, Giri Narasimhan
  Nature Machine Intelligence, 5(9) (IF=25.89)
  

Professional Services

• Program Committee Member
• AAAI Conference on Artificial Intelligence (AAAI), 2026
• The Web Conference (WWW) Web4Good, 2026
• International Joint Conference on Artificial Intelligence (IJCAI), 2025, 2026
• European Conference on Artificial Intelligence (ECAI), 2025
• European Conference on Machine Learning (ECML), 2023, 2024, 2025
• International Conference on Information and Knowledge Management (CIKM), 2026
• Paper Reviewer
• Nature Communications, 2026
• npj Climate and Atmospheric Science, 2026
• Conference on Neural Information Processing Systems (NeurIPS), 2026
• International Conference on Learning Representations (ICLR), 2025, 2026
• SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 2024, 2026, 2027
• NeurIPS Workshop: Tackling Climate Change with Machine Learning, 2024
• ICLR Workshop: Tackling Climate Change with Machine Learning, 2024, 2025
• Transactions on Knowledge Discovery from Data (TKDD), 2026
• IEEE Transactions on Big Data, 2025
• Journal of Intelligent & Fuzzy Systems, 2021, 2022
• Volunteer
• Helped to organize the International Collegiate Programming Contest (ICPC) Southeast USA Regional Contest at Florida International University (2022, 2023, 2024)
• Helped to organize the High School Programming Contest at Florida International University (2025)