← Search

Bobak Kiani

10 accepted papers

2026

Unitary Convolutions for Message-passing and Positional Encodings on Directed Graphs

ICML 2026poster

In many real-world networks, relationships are inherently directional, yet most graph neural networks (GNNs) assume undirected edges, and naïve adaptations of undirected GNNs to directed graphs amplify oversmoothing and gradient pathologies that cap model depth. Unitary graph convolutions (UniConv) …

Cited by 0SourceScholar
2024

On the hardness of learning under symmetries

ICLR 2024spotlight

We study the problem of learning equivariant neural networks via gradient descent. The incorporation of known symmetries ("equivariance") into neural nets has empirically improved the performance of learning pipelines, in domains ranging from biology to computer vision. However, a rich yet separate…

Cited by 13SourcePDFScholar
2023

Self-Supervised Learning with Lie Symmetries for Partial Differential Equations

NeurIPS 2023poster

Machine learning for differential equations paves the way for computationally efficient alternatives to numerical solvers, with potentially broad impacts in science and engineering. Though current algorithms typically require simulated training data tailored to a given setting, one may instead wish…

2023

The SSL Interplay: Augmentations, Inductive Bias, and Generalization

ICML 2023poster

Self-supervised learning (SSL) has emerged as a powerful framework to learn representations from raw data without supervision. Yet in practice, engineers face issues such as instability in tuning optimizers and collapse of representations during training. Such challenges motivate the need for a theo…

Cited by 41SourcePDFScholar
2022

projUNN: efficient method for training deep networks with unitary matrices

NeurIPS 2022accept

In learning with recurrent or very deep feed-forward networks, employing unitary matrices in each layer can be very effective at maintaining long-range stability. However, restricting network parameters to be unitary typically comes at the cost of expensive parameterizations or increased training ru…

2021

Adversarial Robustness Guarantees for Random Deep Neural Networks

ICML 2021spotlight

The reliability of deep learning algorithms is fundamentally challenged by the existence of adversarial examples, which are incorrectly classified inputs that are extremely close to a correctly classified input. We explore the properties of adversarial examples for deep neural networks with random w…