Amir Ziashahabi
publications.md

Publications

Reject Only Critical Tokens: Pivot-Aware Speculative Decoding

Reject Only Critical Tokens: Pivot-Aware Speculative Decoding

accepted
Amir Ziashahabi*, Yavuz Faruk Bakman*, Duygu Nur Yaldiz, Mostafa El-Khamy, Sai Praneeth Karimireddy, Salman Avestimehr
NeurIPS 2025 Efficient Reasoning Workshop · 2025
PaperCode
MobiZO: Enabling Efficient LLM Fine-Tuning at the Edge via Inference Engines

MobiZO: Enabling Efficient LLM Fine-Tuning at the Edge via Inference Engines

accepted
Lei Gao*, Amir Ziashahabi*, Yue Niu, Salman Avestimehr, Murali Annavaram
Conference on Empirical Methods in Natural Language Processing (EMNLP) · 2025
PaperCode
GeoToken: Hierarchical Geolocalization of Images via Next Token Prediction

GeoToken: Hierarchical Geolocalization of Images via Next Token Prediction

accepted
Narges Ghasemi*, Amir Ziashahabi*, Salman Avestimehr, Cyrus Shahabi
IEEE International Conference on Data Mining (ICDM) · 2025
PaperCode
OSMGen: Highly Controllable Satellite Image Synthesis using OpenStreetMap Data

OSMGen: Highly Controllable Satellite Image Synthesis using OpenStreetMap Data

accepted
Amir Ziashahabi*, Narges Ghasemi*, Sajjad Shahabi, John Krumm, Salman Avestimehr, Cyrus Shahabi
NeurIPS 2025 UrbanAI Workshop · 2025
PaperCode
Frequency Domain Diffusion Model with Scale-Dependent Noise Schedule

Frequency Domain Diffusion Model with Scale-Dependent Noise Schedule

published
Amir Ziashahabi, Baturalp Buyukates, Artan Sheshmani, Yi-Zhuang You, Salman Avestimehr
IEEE International Symposium on Information Theory (ISIT) · 2024
PaperCode
Hawk: Accurate and Fast Privacy-Preserving Machine Learning Using Secure Lookup Table Computation

Hawk: Accurate and Fast Privacy-Preserving Machine Learning Using Secure Lookup Table Computation

published
Hamza Saleem, Amir Ziashahabi, Muhammad Naveed, Salman Avestimehr
Privacy Enhancing Technologies Symposium (PETS) · 2024
Paper
Renormalization Group flow, Optimal Transport and Diffusion-based Generative Model

Renormalization Group flow, Optimal Transport and Diffusion-based Generative Model

published
Artan Sheshmani, Yi-Zhuang You, Baturalp Buyukates, Amir Ziashahabi, Salman Avestimehr
Physical Review E · 2024
Paper
PyTorch RPC: Distributed Deep Learning Built on Tensor-Optimized Remote Procedure Calls

PyTorch RPC: Distributed Deep Learning Built on Tensor-Optimized Remote Procedure Calls

published
Shen Li, Pritam Damania, Luca Wehrstedt, et al., including Amir Ziashahabi
Proceedings of Machine Learning and Systems (MLSys) · 2023
Paper
Understanding the Communication Backends of Federated Learning

Understanding the Communication Backends of Federated Learning

preprint
Amir Ziashahabi, Chaoyang He, Salman Avestimehr
Under Review · —