← Search

Keith G. Mills

9 accepted papers

2025

Qua2SeDiMo: Quantifiable Quantization Sensitivity of Diffusion Models

AAAI 2025technical

Diffusion Models (DM) have democratized AI image generation through an iterative denoising process. Quantization is a major technique to alleviate the inference cost and reduce the size of DM denoiser networks. However, as denoisers evolve from variants of convolutional U-Nets toward newer Transform…

Cited by 0SourcePDFScholar
2025

Re-ttention: Ultra Sparse Visual Generation via Attention Statistical Reshape

NeurIPS 2025poster

Diffusion Transformers (DiT) have become the de-facto model for generating high-quality visual content like videos and images. A huge bottleneck is the attention mechanism where complexity scales quadratically with resolution and video length. One logical way to lessen this burden is sparse attentio…

Cited by 0SourcecodeScholar
2024

Building Optimal Neural Architectures using Interpretable Knowledge

CVPR 2024poster

Neural Architecture Search is a costly practice. The fact that a search space can span a vast number of design choices with each architecture evaluation taking nontrivial overhead makes it hard for an algorithm to sufficiently explore candidate networks. In this paper we propose AutoBuild a scheme w…

2024

EiG-Search: Generating Edge-Induced Subgraphs for GNN Explanation in Linear Time

ICML 2024poster

Understanding and explaining the predictions of Graph Neural Networks (GNNs), is crucial for enhancing their safety and trustworthiness. Subgraph-level explanations are gaining attention for their intuitive appeal. However, most existing subgraph-level explainers face efficiency challenges in explai…

2024

GOAt: Explaining Graph Neural Networks via Graph Output Attribution

ICLR 2024poster

Understanding the decision-making process of Graph Neural Networks (GNNs) is crucial to their interpretability. Most existing methods for explaining GNNs typically rely on training auxiliary models, resulting in the explanations remain black-boxed. This paper introduces Graph Output Attribution (GOA…

2023

AIO-P: Expanding Neural Performance Predictors beyond Image Classification

AAAI 2023technical

Evaluating neural network performance is critical to deep neural network design but a costly procedure. Neural predictors provide an efficient solution by treating architectures as samples and learning to estimate their performance on a given task. However, existing predictors are task-dependent, pr…

2023

AutoGO: Automated Computation Graph Optimization for Neural Network Evolution

NeurIPS 2023poster

Optimizing Deep Neural Networks (DNNs) to obtain high-quality models for efficient real-world deployment has posed multi-faceted challenges to machine learning engineers. Existing methods either search for neural architectures in heuristic design spaces or apply low-level adjustments to computation…

2023

GENNAPE: Towards Generalized Neural Architecture Performance Estimators

AAAI 2023technical

Predicting neural architecture performance is a challenging task and is crucial to neural architecture design and search. Existing approaches either rely on neural performance predictors which are limited to modeling architectures in a predefined design space involving specific sets of operators and…

2022

R5: Rule Discovery with Reinforced and Recurrent Relational Reasoning

ICLR 2022spotlight

Systematicity, i.e., the ability to recombine known parts and rules to form new sequences while reasoning over relational data, is critical to machine intelligence. A model with strong systematicity is able to train on small-scale tasks and generalize to large-scale tasks. In this paper, we propose…