← Search

Michael Diskin

5 accepted papers

2023

A critical look at the evaluation of GNNs under heterophily: Are we really making progress?

ICLR 2023poster

Node classification is a classical graph representation learning task on which Graph Neural Networks (GNNs) have recently achieved strong results. However, it is often believed that standard GNNs only work well for homophilous graphs, i.e., graphs where edges tend to connect nodes of the same class.…

2023

SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient

ICML 2023poster

Many deep learning applications benefit from using large models with billions of parameters. Training these models is notoriously expensive due to the need for specialized HPC clusters. In this work, we consider alternative setups for training large models: using cheap ``preemptible'' instances or p…

2022

Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guarantees

NeurIPS 2022accept

Variational inequalities in general and saddle point problems in particular are increasingly relevant in machine learning applications, including adversarial learning, GANs, transport and robust optimization. With increasing data and problem sizes necessary to train high performing models across var…

Cited by 21SourcePDFScholar
2022

Secure Distributed Training at Scale

ICML 2022spotlight

Many areas of deep learning benefit from using increasingly larger neural networks trained on public data, as is the case for pre-trained models for NLP and computer vision. Training such models requires a lot of computational resources (e.g., HPC clusters) that are not available to small research g…

2021

Distributed Deep Learning In Open Collaborations

NeurIPS 2021poster

Modern deep learning applications require increasingly more compute to train state-of-the-art models. To address this demand, large corporations and institutions use dedicated High-Performance Computing clusters, whose construction and maintenance are both environmentally costly and well beyond the…

Cited by 62SourcePDFScholar