activity
20182023
most citedGraphSAINT: Graph Sampling Based Inductive Learning Method

328 citations · 616 across the 23 of their papers we have counts for

collaborators

23 papers

cs.AR2023★ 4 cited

Reuse Kernels or Activations? A Flexible Dataflow for Low-latency Spectral CNN Acceleration

Yue Niu, Rajgopal Kannan, Ajitesh Srivastava +1

Spectral-domain CNNs have been shown to be more efficient than traditional spatial CNNs in terms of reducing computation complexity. However they come with a `kernel explosion' pro…

cs.SI2023

Nowcasting Temporal Trends Using Indirect Surveys

Ajitesh Srivastava, Juan Marcos Ramírez, Sergio Díaz-Aranda +4

Indirect surveys, in which respondents provide information about other people they know, have been proposed for estimating (nowcasting) the size of a \emph{hidden population} where…

q-bio.NC2022

Spatio-Temporal Attention in Multi-Granular Brain Chronnectomes for Detection of Autism Spectrum Disorder

James Orme-Rogers, Ajitesh Srivastava

The traditional methods for detecting autism spectrum disorder (ASD) are expensive, subjective, and time-consuming, often taking years for a diagnosis, with many children growing w…

stat.AP2022★ 2 cited

Shape-based Evaluation of Epidemic Forecasts

Ajitesh Srivastava, Satwant Singh, Fiona Lee

Infectious disease forecasting for ongoing epidemics has been traditionally performed, communicated, and evaluated as numerical targets - 1, 2, 3, and 4 week ahead cases, deaths, a…

stat.ME2022

Robust Scenario Interpretation from Multi-model Prediction Efforts

Yuanhao Lu, Ajitesh Srivastava

Multi-model prediction efforts in infectious disease modeling and climate modeling involve multiple teams independently producing projections under various scenarios. Often these s…

q-bio.PE2022

The Variations of SIkJalpha Model for COVID-19 Forecasting and Scenario Projections

Ajitesh Srivastava

We proposed the SIkJalpha model at the beginning of the COVID-19 pandemic (early 2020). Since then, as the pandemic evolved, more complexities were added to capture crucial factors…

cs.LG2022

Random Forest of Epidemiological Models for Influenza Forecasting

Majd Al Aawar, Ajitesh Srivastava

Forecasting the hospitalizations caused by the Influenza virus is vital for public health planning so that hospitals can be better prepared for an influx of patients. Many forecast…

cs.AR2022★ 3 cited

TransforMAP: Transformer for Memory Access Prediction

Pengmiao Zhang, Ajitesh Srivastava, Anant V. Nori +2

Data Prefetching is a technique that can hide memory latency by fetching data before it is needed by a program. Prefetching relies on accurate memory access prediction, to which ta…

cs.AR2022★ 27 cited

Fine-Grained Address Segmentation for Attention-Based Variable-Degree Prefetching

Pengmiao Zhang, Ajitesh Srivastava, Anant V. Nori +2

Machine learning algorithms have shown potential to improve prefetching performance by accurately predicting future memory accesses. Existing approaches are based on the modeling o…

cs.LG2022★ 54 cited

Decoupling the Depth and Scope of Graph Neural Networks

Hanqing Zeng, Muhan Zhang, Yinglong Xia +6

State-of-the-art Graph Neural Networks (GNNs) have limited scalability with respect to the graph and model sizes. On large graphs, increasing the model depth often means exponentia…

cs.LG2021★ 61 cited

Accelerating Large Scale Real-Time GNN Inference using Channel Pruning

Hongkuan Zhou, Ajitesh Srivastava, Hanqing Zeng +2

Graph Neural Networks (GNNs) are proven to be powerful models to generate node embedding for downstream applications. However, due to the high computation complexity of GNN inferen…

cs.LG2021

The EpiBench Platform to Propel AI/ML-based Epidemic Forecasting: A Prototype Demonstration Reaching Human Expert-level Performance

Ajitesh Srivastava, Tianjian Xu, Viktor K. Prasanna

During the COVID-19 pandemic, a significant effort has gone into developing ML-driven epidemic forecasting techniques. However, benchmarks do not exist to claim if a new AI/ML tech…

cs.LG2020★ 13 cited

Deep Graph Neural Networks with Shallow Subgraph Samplers

Hanqing Zeng, Muhan Zhang, Yinglong Xia +6

While Graph Neural Networks (GNNs) are powerful models for learning representations on graphs, most state-of-the-art models do not have significant accuracy gain beyond two to thre…

cs.LG2020★ 11 cited

Accurate, Efficient and Scalable Training of Graph Neural Networks

Hanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava +2

Graph Neural Networks (GNNs) are powerful deep learning models to generate node embeddings on graphs. When applying deep GNNs on large graphs, it is still challenging to perform tr…

q-bio.PE2020★ 18 cited

Fast and Accurate Forecasting of COVID-19 Deaths Using the SIkJ Model

Ajitesh Srivastava, Tianjian Xu, Viktor K. Prasanna

Forecasting the effect of COVID-19 is essential to design policies that may prepare us to handle the pandemic. Many methods have already been proposed, particularly, to forecast re…

q-bio.PE2020★ 4 cited

Data-driven Identification of Number of Unreported Cases for COVID-19: Bounds and Limitations

Ajitesh Srivastava, Viktor K. Prasanna

Accurate forecasts for COVID-19 are necessary for better preparedness and resource management. Specifically, deciding the response over months or several months requires accurate l…

q-bio.PE2020★ 11 cited

Learning to Forecast and Forecasting to Learn from the COVID-19 Pandemic

Ajitesh Srivastava, Viktor K. Prasanna

Accurate forecasts of COVID-19 is central to resource management and building strategies to deal with the epidemic. We propose a heterogeneous infection rate model with human mobil…

cs.PF2020

Towards High Performance, Portability, and Productivity: Lightweight Augmented Neural Networks for Performance Prediction

Ajitesh Srivastava, Naifeng Zhang, Rajgopal Kannan +1

Writing high-performance code requires significant expertise in the programming language, compiler optimizations, and hardware knowledge. This often leads to poor productivity and…

cs.CV2019★ 2 cited

SPEC2: SPECtral SParsE CNN Accelerator on FPGAs

Yue Niu, Hanqing Zeng, Ajitesh Srivastava +4

To accelerate inference of Convolutional Neural Networks (CNNs), various techniques have been proposed to reduce computation redundancy. Converting convolutional layers into freque…

cs.NI2019

Optimizing Inter-Datacenter Tail Flow Completion Times using Best Worst-case Routing

Max Noormohammadpour, Ajitesh Srivastava, Cauligi S. Raghavendra

Flow routing over inter-datacenter networks is a well-known problem where the network assigns a path to a newly arriving flow potentially according to the network conditions and th…

cs.LG2019★ 328 cited

GraphSAINT: Graph Sampling Based Inductive Learning Method

Hanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava +2

Graph Convolutional Networks (GCNs) are powerful models for learning representations of attributed graphs. To scale GCNs to large graphs, state-of-the-art methods use various layer…

cs.LG2018★ 74 cited

Accurate, Efficient and Scalable Graph Embedding

Hanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava +2

The Graph Convolutional Network (GCN) model and its variants are powerful graph embedding tools for facilitating classification and clustering on graphs. However, a major challenge…

cs.NI2018★ 4 cited

On Minimizing the Completion Times of Long Flows over Inter-Datacenter WAN

Mohammad Noormohammadpour, Ajitesh Srivastava, Cauligi S. Raghavendra

Long flows contribute huge volumes of traffic over inter-datacenter WAN. The Flow Completion Time (FCT) is a vital network performance metric that affects the running time of distr…