← Search

Marinka Zitnik

37 accepted papers

2026

Adaptive Time Series Reasoning via Segment Selection

ICML 2026poster

Time series reasoning tasks increasingly start from a natural language question and require targeted analysis of time series. Relevant evidence may be global or confined to a few short segments, so the model must decide what to inspect. Most existing methods compress the full series into a fixed rep…

Cited by 0SourceScholar
2026

Controllable Sequence Editing for Biological and Clinical Trajectories

ICLR 2026poster

Conditional generation models for longitudinal sequences can produce new or modified trajectories given a conditioning input. However, they often lack control over when the condition should take effect (timing) and which variables it should influence (scope). Most methods either operate only on univ…

Cited by 0SourcecodeScholar
2026

Exploring Relational Reasoning Capabilities in LLMs with REL

ICML 2026poster

Relational reasoning is the ability to infer relations that jointly bind multiple entities, attributes, or variables. While this capability is essential for scientific reasoning, most existing evaluations of relational reasoning in large language models focus on structured inputs such as tables, gra…

Cited by 0SourceScholar
2026

Greater than the Sum of Its Parts: Building Substructure into Protein Encoding Models

ICLR 2026poster

Protein representation learning has achieved major advances using large sequence and structure datasets, yet current models primarily operate at the level of individual residues or entire proteins. This overlooks a critical aspect of protein biology: proteins are composed of recurrent, evolutionaril…

Cited by 0SourceScholar
2026

KnowGuard: Knowledge-Driven Abstention for Multi-Round Clinical Reasoning

ICLR 2026poster

In clinical practice, physicians refrain from making decisions when patient information is insufficient. This behavior, known as abstention, is a critical safety mechanism preventing potentially harmful misdiagnoses. Recent investigations have reported the application of large language models (LLMs)…

Cited by 0SourceScholar
2026

Protein Structure Tokenization via Geometric Byte Pair Encoding

ICLR 2026poster

Protein structure is central to biological function, and enabling multimodal protein models requires joint reasoning over sequence, structure, and function. A key barrier is the lack of principled protein structure tokenizers (PSTs): existing approaches fix token size or rely on continuous vector co…

Cited by 0SourcecodeScholar
2026

Repurposing Foundation Model for Generalizable Medical Time Series Classification

ICLR 2026poster

Medical time series (MedTS) classification suffers from poor generalizability in real-world deployment due to inter- and intra-dataset heterogeneity, such as varying numbers of channels, signal lengths, task definitions, and patient characteristics. % implicit patient characteristics, variable chann…

Cited by 0SourcecodeScholar
2026

SPATIA: Multimodal Generation and Prediction of Spatial Cell Phenotypes

ICML 2026poster

Understanding how cellular morphology, gene expression, and spatial context jointly shape tissue function is a central challenge in biology. Image-based spatial transcriptomics technologies now provide high-resolution measurements of cell images and gene expression profiles, but existing methods typ…

Cited by 0SourceScholar
2025

Composable Interventions for Language Models

ICLR 2025poster

Test-time interventions for language models can enhance factual accuracy, mitigate harmful outputs, and improve model efficiency without costly retraining. But despite a flood of new methods, different types of interventions are largely developing independently. In practice, multiple interventions m…

2025

DQVis Dataset: Natural Language to Biomedical Visualization

NeurIPS 2025poster

Biomedical research data portals are essential resources for scientific inquiry, and interactive exploratory visualizations are an integral component for querying such data repositories. Increasingly, machine learning is being integrated into visualization systems to create natural language interfac…

Cited by 0SourcecodeScholar
2025

Evolutionary Reasoning Does Not Arise in Standard Usage of Protein Language Models

NeurIPS 2025poster

Protein language models (PLMs) are often assumed to capture evolutionary information by training on large protein sequence datasets. Yet it remains unclear whether PLMs can reason about evolution—that is, infer evolutionary relationships between sequences. We test this capability by evaluating wheth…

Cited by 0SourceScholar
2025

KGARevion: An AI Agent for Knowledge-Intensive Biomedical QA

ICLR 2025poster

Biomedical reasoning integrates structured, codified knowledge with tacit, experience-driven insights. Depending on the context, quantity, and nature of available evidence, researchers and clinicians use diverse strategies, including rule-based, prototype-based, and case-based reasoning. Effective m…

Cited by 0SourcePDFScholar
2025

Multimodal Medical Code Tokenizer

ICML 2025poster

Foundation models trained on patient electronic health records (EHRs) require tokenizing medical data into sequences of discrete vocabulary items. Existing tokenizers treat medical codes from EHRs as isolated textual tokens. However, each medical code is defined by its textual description, its posit…

Cited by 4SourcePDFScholar
2025

PyTDC: A multimodal machine learning training, evaluation, and inference platform for biomedical foundation models

ICML 2025poster

Existing biomedical benchmarks do not provide end-to-end infrastructure for training, evaluation, and inference of models that integrate multimodal biological data and a broad range of machine learning tasks in therapeutics. We present PyTDC, an open-source machine-learning platform providing stream…

2024

Generalized Protein Pocket Generation with Prior-Informed Flow Matching

NeurIPS 2024spotlight

Designing ligand-binding proteins, such as enzymes and biosensors, is essential in bioengineering and protein biology. One critical step in this process involves designing protein pockets, the protein interface binding with the ligand. Current approaches to pocket generation often suffer from time-i…

Cited by 2SourcePDFScholar
2024

Graph Adversarial Diffusion Convolution

ICML 2024poster

This paper introduces a min-max optimization formulation for the Graph Signal Denoising (GSD) problem. In this formulation, we first maximize the second term of GSD by introducing perturbations to the graph structure based on Laplacian distance and then minimize the overall loss of the GSD. By solvi…

2024

Invariant Tokenization of Crystalline Materials for Language Model Enabled Generation

NeurIPS 2024poster

We consider the problem of crystal materials generation using language models (LMs). A key step is to convert 3D crystal structures into 1D sequences to be processed by LMs. Prior studies used the crystallographic information framework (CIF) file stream, which fails to ensure SE(3) and periodic inva…

Cited by 4SourcePDFScholar
2024

Let's Think Outside the Box: Exploring Leap-of-Thought in Large Language Models with Creative Humor Generation

CVPR 2024poster

Chain-of-Thought (CoT) guides large language models (LLMs) to reason step-by-step and can motivate their logical reasoning ability. While effective for logical tasks CoT is not conducive to creative problem-solving which often requires out-of-box thoughts and is crucial for innovation advancements.…

2024

Position: TrustLLM: Trustworthiness in Large Language Models

ICML 2024poster

Large language models (LLMs) have gained considerable attention for their excellent natural language processing capabilities. Nonetheless, these LLMs present many challenges, particularly in the realm of trustworthiness. This paper introduces TrustLLM, a comprehensive study of trustworthiness in LLM…

Cited by 95SourcePDFScholar
2024

UniTS: A Unified Multi-Task Time Series Model

NeurIPS 2024poster

Although pre-trained transformers and reprogrammed text-based LLMs have shown strong performance on time series tasks, the best-performing architectures vary widely across tasks, with most models narrowly focused on specific areas, such as time series forecasting. Unifying predictive and generative…

2023

Domain Adaptation for Time Series Under Feature and Label Shifts

ICML 2023poster

Unsupervised domain adaptation (UDA) enables the transfer of models trained on source domains to unlabeled target domains. However, transferring complex time series models presents challenges due to the dynamic temporal structure variations across domains. This leads to feature shifts in the time an…

2023

Encoding Time-Series Explanations through Self-Supervised Model Behavior Consistency

NeurIPS 2023spotlight

Interpreting time series models is uniquely challenging because it requires identifying both the location of time series signals that drive model predictions and their matching to an interpretable temporal pattern. While explainers from other modalities can be applied to time series, their inductive…

2023

Full-Atom Protein Pocket Design via Iterative Refinement

NeurIPS 2023spotlight

The design of \emph{de novo} functional proteins that bind with specific ligand molecules is crucial in various domains like therapeutics and bio-engineering. One vital yet challenging step is to design the protein pocket, the cavity region of protein where the ligand binds with. Existing methods su…

2023

GNNDelete: A General Strategy for Unlearning in Graph Neural Networks

ICLR 2023poster

Graph unlearning, which involves deleting graph elements such as nodes, node labels, and relationships from a trained graph neural network (GNN) model, is crucial for real-world applications where data elements may become irrelevant, inaccurate, or privacy-sensitive. However, existing methods for gr…

2022

Graph-Guided Network for Irregularly Sampled Multivariate Time Series

ICLR 2022poster

In many domains, including healthcare, biology, and climate science, time series are irregularly sampled with varying time intervals between successive readouts and different subsets of variables (sensors) observed at different time points. Here, we introduce RAINDROP, a graph neural network that em…

2022

OpenXAI: Towards a Transparent Evaluation of Model Explanations

NeurIPS 2022accept

While several types of post hoc explanation methods have been proposed in recent literature, there is very little work on systematically benchmarking these methods. Here, we introduce OpenXAI, a comprehensive and extensible open-source framework for evaluating and benchmarking post hoc explanation m…

2022

Probing GNN Explainers: A Rigorous Theoretical and Empirical Analysis of GNN Explanation Methods

AISTATS 2022poster

As Graph Neural Networks (GNNs) are increasingly being employed in critical real-world applications, several methods have been proposed in recent literature to explain the predictions of these models. However, there has been little to no work on systematically analyzing the reliability of these meth…

Cited by 77SourcePDFScholar
2022

Self-Supervised Contrastive Pre-Training For Time Series via Time-Frequency Consistency

NeurIPS 2022accept

Pre-training on time series poses a unique challenge due to the potential mismatch between pre-training and target domains, such as shifts in temporal dynamics, fast-evolving trends, and long-range and short-cyclic effects, which can lead to poor downstream performance. While domain adaptation metho…

2021

Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Development

NeurIPS 2021poster

Therapeutics machine learning is an emerging field with incredible opportunities for innovation and impact. However, advancement in this field requires the formulation of meaningful tasks and careful curation of datasets. Here, we introduce Therapeutics Data Commons (TDC), the first unifying platfor…

Cited by 354SourcecodeScholar
2021

Towards a unified framework for fair and stable graph representation learning

UAI 2021poster

As the representations output by Graph Neural Networks (GNNs) are increasingly employed in real-world applications, it becomes important to ensure that these representations are fair and stable. In this work, we establish a key connection between counterfactual fairness and stability and leverage it…

2020

Open Graph Benchmark: Datasets for Machine Learning on Graphs

NeurIPS 2020spotlight

We present the Open Graph Benchmark (OGB), a diverse set of challenging and realistic benchmark datasets to facilitate scalable, robust, and reproducible graph machine learning (ML) research. OGB datasets are large-scale, encompass multiple important graph ML tasks, and cover a diverse range of doma…

2020

Strategies for Pre-training Graph Neural Networks

ICLR 2020spotlight

Many applications of machine learning require a model to make accurate pre-dictions on test examples that are distributionally different from training ones, while task-specific labels are scarce during training. An effective approach to this challenge is to pre-train a model on related tasks where d…

Cited by 1823SourcecodeScholar
2019

GNNExplainer: Generating Explanations for Graph Neural Networks

NeurIPS 2019poster

Graph Neural Networks (GNNs) are a powerful tool for machine learning on graphs.GNNs combine node feature information with the graph structure by recursively passing neural messages along edges of the input graph. However, incorporating both graph structure and feature information leads to complex…

2018

Embedding Logical Queries on Knowledge Graphs

NeurIPS 2018poster

Learning low-dimensional embeddings of knowledge graphs is a powerful approach used to predict unobserved or missing edges between entities. However, an open challenge in this area is developing techniques that can go beyond simple edge prediction and handle more complex logical queries, which might…