← Search

Amir Yazdanbakhsh

19 accepted papers

2026

ParEVO: Synthesizing Code for Irregular Data: High-Performance Parallelism through Agentic Evolution

ICML 2026poster

The transition from sequential to parallel computing is essential for modern high-performance applications but is hindered by the steep learning curve of concurrent programming. This challenge is magnified for \textbf{irregular data structures} (such as sparse graphs, unbalanced trees, and non-unifo…

Cited by 0SourceScholar
2026

QuArch: A Benchmark for Evaluating LLM Reasoning in Computer Architecture

ICML 2026poster

The field of computer architecture, which bridges high-level software abstractions and low-level hardware implementations, remains absent from current large language model (LLM) evaluations. To this end, we present QuArch (pronounced 'quark'), the first benchmark designed to facilitate the developme…

Cited by 0SourceScholar
2026

SWE-fficiency: Can Language Models Optimize Real-World Repositories on Real Workloads?

ICML 2026poster

Optimizing the performance of large-scale software repositories demands expertise in code reasoning and software engineering (SWE) to reduce runtime while preserving program correctness. However, most benchmarks emphasize what to fix rather than how to fix code. We introduce SWE-fficiency, a benchma…

Cited by 0SourceScholar
2025

Effective Interplay between Sparsity and Quantization: From Theory to Practice

ICLR 2025spotlight

The increasing size of deep neural networks (DNNs) necessitates effective model compression to reduce their computational and memory footprints. Sparsity and quantization are two prominent compression methods that have been shown to reduce DNNs' computational and memory footprints significantly whil…

Cited by 5SourcePDFScholar
2025

Learning to Keep a Promise: Scaling Language Model Decoding Parallelism with Learned Asynchronous Decoding

ICML 2025poster

Decoding with autoregressive language models traditionally occurs sequentially, generating one token after another. Recent attempts to introduce parallelism require a pre-determined structure in the generated content to implement parallel generation, such as by pattern-matching on bullet points. In…

Cited by 0SourcePDFScholar
2025

SLiM: One-shot Quantization and Sparsity with Low-rank Approximation for LLM Weight Compression

ICML 2025poster

Conventional model compression techniques for LLMs address high memory consumption and slow inference challenges but typically require computationally expensive retraining to preserve accuracy. In contrast, one-shot compression methods eliminate retraining cost, but struggle to achieve accuracy comp…

2025

SLoPe: Double-Pruned Sparse Plus Lazy Low-Rank Adapter Pretraining of LLMs

ICLR 2025poster

We propose SLoPe, a Double-Pruned **S**parse Plus **L**azy L**o**w-rank Adapter **P**r**e**training method for LLMs that improves the accuracy of sparse LLMs while accelerating their pretraining and inference and reducing their memory footprint. Sparse pretraining of LLMs reduces the accuracy of the…

2025

SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity

ICML 2025poster

Fine-tuning LLMs is both computationally and memory-intensive. While parameter-efficient fine-tuning methods, such as QLoRA and DoRA, reduce the number of trainable parameters and lower memory usage, they do not decrease computational cost. In some cases, they may even slow down fine-tuning. In this…

Cited by 0SourcePDFScholar
2025

The Journey Matters: Average Parameter Count over Pre-training Unifies Sparse and Dense Scaling Laws

ICLR 2025poster

Pruning eliminates unnecessary parameters in neural networks; it offers a promising solution to the growing computational demands of large language models (LLMs). While many focus on post-training pruning, sparse pre-training--which combines pruning and pre-training into a single phase--provides a…

Cited by 0SourcePDFScholar
2024

CodeRosetta: Pushing the Boundaries of Unsupervised Code Translation for Parallel Programming

NeurIPS 2024poster

Automatic translation of programming languages has garnered renewed interest, driven by recent advancements in large language models (LLMs). Encoder-decoder transformer models, in particular, have shown promise in translating between different programming languages. However, translating between a la…

Cited by 1SourcePDFScholar
2024

Learning Performance-Improving Code Edits

ICLR 2024spotlight

With the decline of Moore's law, optimizing program performance has become a major focus of software research. However, high-level optimizations such as API and algorithm changes remain elusive due to the difficulty of understanding the semantics of code. Simultaneously, pretrained large language mo…

2024

ShiftAddLLM: Accelerating Pretrained LLMs via Post-Training Multiplication-Less Reparameterization

NeurIPS 2024poster

Large language models (LLMs) have shown impressive performance on language tasks but face challenges when deployed on resource-constrained devices due to their extensive parameters and reliance on dense multiplications, resulting in high memory demands and latency bottlenecks. Shift-and-add reparame…

2024

USM-Lite: Quantization and Sparsity Aware Fine-Tuning for Speech Recognition with Universal Speech Models

ICASSP 2024accepted

End-to-end automatic speech recognition (ASR) models have seen revolutionary quality gains with the recent development of large-scale universal speech models (USM). However, deploying these massive USMs is extremely expensive due to the enormous memory usage and computational cost. Therefore, model…

Cited by 0SourceScholar
2024

When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models

ICML 2024poster

Autoregressive Large Language Models (LLMs) have achieved impressive performance in language tasks but face two significant bottlenecks: (1) quadratic complexity in the attention module as the number of tokens increases, and (2) limited efficiency due to the sequential processing nature of autoregre…

2023

STEP: Learning N:M Structured Sparsity Masks from Scratch with Precondition

ICML 2023poster

Recent innovations on hardware (e.g. Nvidia A100) have motivated learning N:M structured sparsity masks from scratch for fast model inference. However, state-of-the-art learning recipes in this regime (e.g. SR-STE) are proposed for non-adaptive optimizers like momentum SGD, while incurring non-trivi…

Cited by 15SourcePDFScholar
2023

Self-Refine: Iterative Refinement with Self-Feedback

NeurIPS 2023poster

Like humans, large language models (LLMs) do not always generate the best output on their first try. Motivated by how humans refine their written text, we introduce Self-Refine, an approach for improving initial outputs from LLMs through iterative feedback and refinement. The main idea is to generat…

Cited by 1546SourcePDFScholar
2023

What Makes Chain-of-Thought Prompting Effective? A Counterfactual Study

EMNLP 2023long findings

The effectiveness of Chain-of-thought prompting (CoT) has been widely recognized, but the underlying mechanisms behind its success, the reason why it just works for a wide range of tasks, remains an open question. To investigate this, we employ a counterfactual prompting approach, systematically man…

Cited by 49SourceScholar
2022

Data-Driven Offline Optimization for Architecting Hardware Accelerators

ICLR 2022poster

To attain higher efficiency, the industry has gradually reformed towards application-specific hardware accelerators. While such a paradigm shift is already starting to show promising results, designers need to spend considerable manual effort and perform large number of time-consuming simulations to…

2020

Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation

ICLR 2020poster

Achieving faster execution with shorter compilation time can foster further diversity and innovation in neural networks. However, the current paradigm of executing neural networks either relies on hand-optimized libraries, traditional compilation heuristics, or very recently genetic algorithms and o…

Cited by 101SourceScholar