← Search

Krishna Sri Ipsit Mantri

7 accepted papers

2026

Bridging Input Feature Spaces Towards Graph Foundation Models

ICLR 2026poster

Unlike vision and language domains, graph learning lacks a shared input space, as input features differ across graph datasets not only in semantics, but also in value ranges and dimensionality. This misalignment prevents graph models from generalizing across datasets, limiting their use as foundatio…

Cited by 0SourcecodeScholar
2026

FLASH: Flexible Learning of Adaptive Sampling from History in Temporal Graph Neural Networks

IJCAI 2026

Aggregating temporal signals from historic interactions is a key step in future link prediction on dynamic graphs. However, incorporating long histories is resource-intensive. Hence, temporal graph neural networks (TGNNs) often rely on historical neighbors sampling heuristics such as uniform samplin

Cited by 0Scholar
2026

Revisting Node Affinity Prediction In Temporal Graphs

ICLR 2026poster

Node affinity prediction is a common task that is widely used in temporal graph learning with applications in social and financial networks, recommender systems, and more. Recent works have addressed this task by adapting state-of-the-art dynamic link property prediction models to node affinity pre…

Cited by 0SourcecodeScholar
2026

Towards Improved Sentence Representations using Token Graphs

ICLR 2026poster

Obtaining a single-vector representation from a Large Language Model's (LLM) token-level outputs is a critical step for nearly all sentence-level tasks. However, standard pooling methods like mean or max aggregation treat tokens as an independent set, discarding the rich relational structure capture…

Cited by 0SourcecodeScholar
2025

DiTASK: Multi-Task Fine-Tuning with Diffeomorphic Transformations

CVPR 2025poster

Pre-trained Vision Transformers now serve as powerful tools for computer vision. Yet, efficiently adapting them for multiple tasks remains a challenge that arises from the need to modify the rich hidden representations encoded by the learned weight matrices, without inducing interference between tas…

2024

DiGRAF: Diffeomorphic Graph-Adaptive Activation Function

NeurIPS 2024poster

In this paper, we propose a novel activation function tailored specifically for graph data in Graph Neural Networks (GNNs). Motivated by the need for graph-adaptive and flexible activation functions, we introduce DiGRAF, leveraging Continuous Piecewise-Affine Based (CPAB) transformations, which we a…

2023

RWKV: Reinventing RNNs for the Transformer Era

EMNLP 2023long findings

Transformers have revolutionized almost all natural language processing (NLP) tasks but suffer from memory and computational complexity that scales quadratically with sequence length. In contrast, recurrent neural networks (RNNs) exhibit linear scaling in memory and computational requirements but st…

Cited by 0SourceScholar