← Search

Mark Gerstein

12 accepted papers

2026

Dispersion Loss Counteracts Embedding Condensation and Improves Generalization in Small Language Models

ICML 2026poster

Large language models (LLMs) achieve remarkable performance through ever-increasing parameter counts, but scaling incurs steep computational costs. To better understand LLM scaling, we study representational differences between LLMs and their smaller counterparts, with the goal of replicating the re…

Cited by 0SourceScholar
2026

Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learned Force Fields

ICML 2026poster

Generative models for 3D molecular conformations must respect Euclidean symmetries and concentrate probability mass on thermodynamically favorable, mechanically stable structures. However, E(3)-equivariant diffusion models often reproduce biases from semi-empirical training data rather than capturin…

Cited by 0SourceScholar
2025

ChemAgent: Self-updating Memories in Large Language Models Improves Chemical Reasoning

ICLR 2025poster

Chemical reasoning usually involves complex, multi-step processes that demand precise calculations, where even minor errors can lead to cascading failures. Furthermore, large language models (LLMs) encounter difficulties handling domain-specific formulas, executing reasoning steps accurately, and in…

2025

E2Former: An Efficient and Equivariant Transformer with Linear-Scaling Tensor Products

NeurIPS 2025spotlight

Equivariant Graph Neural Networks (EGNNs) have demonstrated significant success in modeling microscale systems, including those in chemistry, biology and materials science. However, EGNNs face substantial computational challenges due to the high cost of constructing edge features via spherical tenso…

Cited by 0SourceScholar
2025

Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systems

ICLR 2025spotlight

Density Functional Theory (DFT) is a pivotal method within quantum chemistry and materials science, with its core involving the construction and solution of the Kohn-Sham Hamiltonian. Despite its importance, the application of DFT is frequently limited by the substantial computational resources requ…

Cited by 0SourcePDFScholar
2025

Med-PRM: Medical Reasoning Models with Stepwise, Guideline-verified Process Rewards

EMNLP 2025

Large language models have shown promise in clinical decision making, but current approaches struggle to localize and correct errors at specific steps of the reasoning process. This limitation is critical in medicine, where identifying and addressing reasoning errors is essential for accurate diagno

Cited by 0SourcePDFScholar
2024

Investigating Data Contamination in Modern Benchmarks for Large Language Models

NAACL 2024long

Recent observations have underscored a disparity between the inflated benchmark scores and the actual performance of LLMs, raising concerns about potential contamination of evaluation benchmarks. This issue is especially critical for closed-source models and certain open-source models where training…

Cited by 101SourcePDFScholar
2024

MIMIR: A Customizable Agent Tuning Platform for Enhanced Scientific Applications

EMNLP 2024system demonstrations

Recently, large language models (LLMs) have evolved into interactive agents, proficient in planning, tool use, and task execution across various tasks. However, without agent-tuning, open-source models like LLaMA2 currently struggle to match the efficiency of larger models such as GPT-4 in scientifi…

2024

MedAgents: Large Language Models as Collaborators for Zero-shot Medical Reasoning

ACL 2024findings

Large language models (LLMs), despite their remarkable progress across various general domains, encounter significant barriers in medicine and healthcare. This field faces unique challenges such as domain-specific terminologies and reasoning over specialized knowledge. To address these issues, we pr…

2024

Struc-Bench: Are Large Language Models Good at Generating Complex Structured Tabular Data?

NAACL 2024short

Despite the remarkable capabilities of Large Language Models (LLMs) like GPT-4, producing complex, structured tabular data remains challenging. Our study assesses LLMs’ proficiency in structuring tables and introduces a novel fine-tuning method, cognizant of data structures, to bolster their perform…

2024

ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIs

ICLR 2024spotlight

Despite the advancements of open-source large language models (LLMs), e.g., LLaMA, they remain significantly limited in tool-use capabilities, i.e., using external tools (APIs) to fulfill human instructions. The reason is that current instruction tuning largely focuses on basic language tasks but ig…

2023

Disentangled Wasserstein Autoencoder for T-Cell Receptor Engineering

NeurIPS 2023poster

In protein biophysics, the separation between the functionally important residues (forming the active site or binding surface) and those that create the overall structure (the fold) is a well-established and fundamental concept. Identifying and modifying those functional sites is critical for protei…

Cited by 5SourcePDFScholar