← Search

Babak Ehteshami Bejnordi

9 accepted papers

2026

Dirichlet-Prior Shaping: Guiding Expert Specialization in Upcycled MoEs

ICML 2026poster

Upcycling pre-trained dense models into sparse Mixture-of-Experts (MoEs) efficiently increases model capacity but often suffers from poor expert specialization due to naive weight replication. We introduce Dirichlet-Prior Shaping Loss (DPSL), a novel router regularization technique that directly sha…

Cited by 0SourceScholar
2024

$\textit{Read-ME}$: Refactorizing LLMs as Router-Decoupled Mixture of Experts with System Co-Design

NeurIPS 2024poster

The proliferation of large language models (LLMs) has led to the adoption of Mixture-of-Experts (MoE) architectures that dynamically leverage specialized subnetworks for improved efficiency and performance. Despite their benefits, MoE models face significant challenges during inference, including in…

2023

Scalarization for Multi-Task and Multi-Domain Learning at Scale

NeurIPS 2023poster

Training a single model on multiple input domains and/or output tasks allows for compressing information from multiple sources into a unified backbone hence improves model efficiency. It also enables potential positive knowledge transfer across tasks/domains, leading to improved accuracy and data-ef…

Cited by 18SourcePDFScholar
2022

SALISA: Saliency-Based Input Sampling for Efficient Video Object Detection

ECCV 2022poster

"High-resolution images are widely adopted for high-performance object detection in videos. However, processing high-resolution inputs comes with high computation costs, and naive down-sampling of the input to reduce the computation costs quickly degrades the detection performance. In this paper, we…

Cited by 21SourcePDFScholar
2021

FrameExit: Conditional Early Exiting for Efficient Video Recognition

CVPR 2021poster

In this paper, we propose a conditional early exiting framework for efficient video recognition. While existing works focus on selecting a subset of salient frames to reduce the computation costs, we propose to use a simple sampling strategy combined with conditional early exiting to enable efficien…

Cited by 102PDFcodeScholar
2021

Skip-Convolutions for Efficient Video Processing

CVPR 2021poster

We propose Skip-Convolutions to leverage the large amount of redundancies in video streams and save computations. Each video is represented as a series of changes across frames and network activations, denoted as residuals. We reformulate standard convolution to be efficiently computed on residual f…

Cited by 62PDFScholar
2020

Batch-shaping for learning conditional channel gated networks

ICLR 2020poster

We present a method that trains large capacity neural networks with significantly improved accuracy and lower dynamic computational cost. This is achieved by gating the deep-learning architecture on a fine-grained-level. Individual convolutional maps are turned on/off conditionally on features in th…

Cited by 84SourceScholar
2020

Conditional Channel Gated Networks for Task-Aware Continual Learning

CVPR 2020oral

Convolutional Neural Networks experience catastrophic forgetting when optimized on a sequence of learning problems: as they meet the objective of the current training examples, their performance on previous tasks drops drastically. In this work, we introduce a novel framework to tackle this problem…

Cited by 270PDFScholar