← Search

Artem Babenko

38 accepted papers

2026

GraphPFN: A Prior-Data Fitted Network for Graph Node-Level Tasks

ICML 2026poster

Graph foundation models face several fundamental challenges including transferability across datasets and data scarcity, which calls into question the very feasibility of graph foundation models. However, despite similar challenges, the tabular domain has recently witnessed the emergence of the firs…

Cited by 0SourceScholar
2026

Scale-wise Distillation of Diffusion Models

ICLR 2026poster

Recent diffusion distillation methods have achieved remarkable progress, enabling high-quality ${\sim}4$-step sampling for large-scale text-conditional image and video diffusion models (DMs). However, further reducing the number of sampling steps becomes more and more challenging, suggesting that e…

Cited by 0SourcecodeScholar
2026

TabPack: Efficient Hyperparameter Ensembles for Tabular Deep Learning

ICML 2026poster

Deep learning models for supervised learning on tabular data are rapidly improving. Notably, ensembles (mixtures of multiple models) often play an important role in achieving top performance, which motivates designing ensemble-first systems rather than treating ensembling as an ad hoc trick. In this…

Cited by 0SourceScholar
2025

TabM: Advancing tabular deep learning with parameter-efficient ensembling

ICLR 2025poster

Deep learning architectures for supervised learning on tabular data range from simple multilayer perceptrons (MLP) to sophisticated Transformers and retrieval-augmented methods. This study highlights a major, yet so far overlooked opportunity for substantially improving tabular MLPs; namely, paramet…

2025

TabReD: Analyzing Pitfalls and Filling the Gaps in Tabular Deep Learning Benchmarks

ICLR 2025spotlight

Advances in machine learning research drive progress in real-world applications. To ensure this progress, it is important to understand the potential pitfalls on the way from a novel method's success on academic benchmarks to its practical deployment. In this work, we analyze existing tabular deep…

2024

Extreme Compression of Large Language Models via Additive Quantization

ICML 2024poster

The emergence of accurate open large language models (LLMs) has led to a race towards performant quantization techniques which can enable their execution on end-user devices. In this paper, we revisit the problem of ``extreme'' LLM compression---defined as targeting extremely low bit counts, such as…

2024

Invertible Consistency Distillation for Text-Guided Image Editing in Around 7 Steps

NeurIPS 2024poster

Diffusion distillation represents a highly promising direction for achieving faithful text-to-image generation in a few sampling steps. However, despite recent successes, existing distilled models still do not provide the full spectrum of diffusion abilities, such as real image inversion, which enab…

Cited by 2SourcePDFScholar
2024

TabR: Tabular Deep Learning Meets Nearest Neighbors

ICLR 2024poster

Deep learning (DL) models for tabular data problems (e.g. classification, regression) are currently receiving increasingly more attention from researchers. However, despite the recent efforts, the non-DL algorithms based on gradient-boosted decision trees (GBDT) remain a strong go-to solution for th…

Cited by 39SourcePDFScholar
2024

Your Student is Better Than Expected: Adaptive Teacher-Student Collaboration for Text-Conditional Diffusion Models

CVPR 2024poster

Knowledge distillation methods have recently shown to be a promising direction to speedup the synthesis of large-scale diffusion models by requiring only a few inference steps. While several powerful distillation methods were recently proposed the overall quality of student samples is typically lowe…

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

Characterizing Graph Datasets for Node Classification: Homophily-Heterophily Dichotomy and Beyond

NeurIPS 2023poster

Homophily is a graph property describing the tendency of edges to connect similar nodes; the opposite is called heterophily. It is often believed that heterophilous graphs are challenging for standard message-passing graph neural networks (GNNs), and much effort has been put into developing efficien…

Cited by 79SourcePDFScholar
2023

Evaluating Robustness and Uncertainty of Graph Models Under Structural Distributional Shifts

NeurIPS 2023poster

In reliable decision-making systems based on machine learning, models have to be robust to distributional shifts or provide the uncertainty of their predictions. In node-level problems of graph learning, distributional shifts can be especially complex since the samples are interdependent. To evaluat…

2023

Is This Loss Informative? Faster Text-to-Image Customization by Tracking Objective Dynamics

NeurIPS 2023poster

Text-to-image generation models represent the next step of evolution in image synthesis, offering a natural way to achieve flexible yet fine-grained control over the result. One emerging area of research is the fast adaptation of large text-to-image models to smaller datasets or new visual concepts.…

2023

TabDDPM: Modelling Tabular Data with Diffusion Models

ICML 2023poster

Denoising diffusion probabilistic models are becoming the leading generative modeling paradigm for many important data modalities. Being the most prevalent in the computer vision community, diffusion models have recently gained some attention in other domains, including speech, NLP, and graph-like d…

Cited by 323SourcePDFScholar
2022

Label-Efficient Semantic Segmentation with Diffusion Models

ICLR 2022poster

Denoising diffusion probabilistic models have recently received much research attention since they outperform alternative approaches, such as GANs, and currently provide state-of-the-art generative performance. The superior performance of diffusion models has made them an appealing tool in several a…

2022

On Embeddings for Numerical Features in Tabular Deep Learning

NeurIPS 2022accept

Recently, Transformer-like deep architectures have shown strong performance on tabular data problems. Unlike traditional models, e.g., MLP, these architectures map scalar values of numerical features to high-dimensional embeddings before mixing them in the main backbone. In this work, we argue that…

2021

Latent Transformations via NeuralODEs for GAN-Based Image Editing

ICCV 2021poster

Recent advances in high-fidelity semantic image editing heavily rely on the presumably disentangled latent spaces of the state-of-the-art generative models, such as StyleGAN. Specifically, recent works show that it is possible to achieve decent controllability of attributes in the face images via li…

Cited by 19PDFcodeScholar
2021

Neural Side-by-Side: Predicting Human Preferences for No-Reference Super-Resolution Evaluation

CVPR 2021poster

Super-resolution based on deep convolutional networks is currently gaining much attention from both academia and industry. However, lack of proper evaluation measures makes it difficult to compare approaches, hampering progress in the field. Traditional measures, such as PSNR or SSIM, are known to p…

Cited by 17PDFcodeScholar
2021

Object Segmentation Without Labels with Large-Scale Generative Models

ICML 2021spotlight

The recent rise of unsupervised and self-supervised learning has dramatically reduced the dependency on labeled data, providing high-quality representations for transfer on downstream tasks. Furthermore, recent works also employed these representations in a fully unsupervised setup for image classif…

2021

Revisiting Deep Learning Models for Tabular Data

NeurIPS 2021poster

The existing literature on deep learning for tabular data proposes a wide range of novel architectures and reports competitive results on various datasets. However, the proposed models are usually not properly compared to each other and existing works often use different benchmarks and experiment pr…

2020

Unsupervised Discovery of Interpretable Directions in the GAN Latent Space

ICML 2020poster

The latent spaces of GAN models often have semantically meaningful directions. Moving in these directions corresponds to human-interpretable image transformations, such as zooming or recoloring, enabling a more controllable generation process. However, the discovery of such directions is currently p…

2019

Beyond Vector Spaces: Compact Data Representation as Differentiable Weighted Graphs

NeurIPS 2019poster

Learning useful representations is a key ingredient to the success of modern machine learning. Currently, representation learning mostly relies on embedding data into Euclidean space. However, recent work has shown that data in some domains is better modeled by non-euclidean metric spaces, and inapp…

2018

Revisiting the Inverted Indices for Billion-Scale Approximate Nearest Neighbors

ECCV 2018poster

This work addresses the problem of billion-scale nearest neighbor search. The state-of-the-art retrieval systems for billion-scale databases are currently based on the inverted multi-index, the recently proposed generalization of the inverted index structure. The multi-index provides a very fine-gra…

2017

Product Split Trees

CVPR 2017poster

In this work, we introduce a new kind of spatial partition trees for efficient nearest-neighbor search. Our approach first identifies a set of useful data splitting directions, and then learns a codebook that can be used to encode such directions. We use the product-quantization idea in order to mak…

Cited by 12PDFScholar