Sujai Hiremath
I am motivated by a curiosity of the world. I work primarily on (1) developing robust and sample efficient causal structure learning algorithms, and their application in (2) reinforcement learning, as well as (3) agentic security and AI safety.
I believe causality will be essential to the efficient and safe development of ML systems that have prediction, control and counterfactual reasoning capabitilies that can generalize to unseen environments. I aim to close the gap between theory and practice to enable AI to truly understand our causal world.
I enjoy research collabs and helping to organize NYC AI Safety meetups - feel free to reach out :)
Google Scholar| Contact | LinkedIn | CV
Drafts
From Causal Structure to Efficient Representations: Deep Reinforcement Learning with Causal Rank Regularization
Sujai Hiremath, Shiva Prasad Kasiviswanathan, Dominik Janzing, Patrick Blöbaum, Elke Kirschbaum, Kyra Gan, Krishna Balasubramanian.
From Detection to Attribution: Identifying Malicious Documents in Prompt Injection Attacks on LLM Agents via Root Cause Analysis
Sujai Hiremath.
Publications
From Guess2Graph: When and How Can Unreliable Experts Safely Boost Causal Discovery in Finite Samples?
Sujai Hiremath, Dominik Janzing, Philipp Faller, Patrick Blöbaum, Elke Kirschbaum, Shiva Prasad Kasiviswanathan, Kyra Gan.
When Additive Noise Meets Unobserved Mediators: Bivariate Denoising Diffusion for Causal Discovery.
Dominik Meier, Sujai Hiremath, Promit Ghosal, Kyra Gan.
LoSAM: Local Search in Additive Noise Models with Mixed Mechanisms and General Noise for Global Causal Discovery.
Sujai Hiremath, Promit Ghosal, Kyra Gan.
Hybrid Top-Down Global Causal Discovery with Local Search for Linear and Nonlinear Additive Noise Models.
Sujai Hiremath, Jacqueline Maasch, Mengxiao Gao, Promit Ghosal, Kyra Gan.
Books, Art, Music, Movies and Places
The Skeptic Way, Benson Mates.
Against Method, Paul Feyerabend.
Ways of Worldmaking, Nelson Goodman.
Prolegomena to Any Future Metaphysics, Immanuel Kant.
The Book of Disquiet, Fernando Pessoa.
Fear and Trembling, Søren Kierkegaard.
Fixed Points, Wassily Kandinsky.
Weeping Willow, Claude Monet.
Women Picking Olives, Vincent van Gogh.
Life Is Long, Brian Eno & David Byrne.
Small Hours, John Martyn.
I Wish I Knew How It Would Feel to Be Free, Nina Simone.
12 Angry Men , Reginald Rose.
The Day He Arrives, Hong Sang-Soo.
The Man Who Sleeps, Georges Perec.
Enter the Void, Gaspar Noé.
Los Angeles, Echo Mountain.
San Francisco, Golden Gate Park.
New York City, Fasan Coffee Shop.
Tübingen, Amazon Science Office.
Freiburg im Breisgau, Vauban District.
Köln, Mensa Zülpicher.
Berlin, Sisyphos Nightclub.
Munich, Lenbachhaus Art Museum.
Copenhagen , Freetown Christiania.
Oslo, Floating Sauna.
Mälmo, Pildammsparken.
Giverny , Monet's House and Gardens.
Paris, Crow Lounge.
Strasbourg, Parc du Contades.
Amsterdam , Ridammerhoeve Goat Farm.
Zandvoort , Zuid-Kennemerland National Park.
Prague, Riegrovy Sady Park.
Vienna, Prater Hauptallee Park.
Venice, Retro Jazz Bar.
Rome, Fassi Gelateria.
Naples, Pildammsparken.
Zürich, ETH Zürich.
Lucerne, Lake Lucerne.
Bern, Kunst Art Museum.
Basel, Kunst Art Museum.
London , Round Pound Lake.
Istanbul , Balat Neighborhood.
Rio di Janeiro , Tijuca National Park.
Vancouver , Whistler Blackcomb Ski Resort.
San Diego , Coronado Beach.