Demo Data from EDBT 2026 / reviewers in the wild / expert
Awni Y. Hannun
dblp:147/5262
· DBLP profile ↗
24ranked-venue papers
3as first author
9since 2021 · last 2022
—ORCID · none
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 20 · 3 first-author · 7 since 2021Graphics, computer vision, multimedia, augmented reality and games · 11 · 2 first-author · 3 since 2021Databases, data management, data science and information retrieval · 1
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
10 papers |
Efficient and distributed learning · 27% Speech recognition and synthesis · 19% Trustworthy machine learning · 18% | |
| Network and information security
3 papers |
Privacy and data protection · 51% Security and privacy of machine learning · 25% Cryptographic protocols and secure computation · 12% | |
| Computer architecture, parallel and distributed computing, and storage systems
2 papers |
GPUs and heterogeneous computing · 77% High-performance computing · 12% Cloud and datacenter computing · 12% | |
| Human-computer interaction and pervasive computing
1 paper |
Health and well-being technologies · 100% |
Topics — the 19 heaviest of 27, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Privacy and data protection
privacy-preserving machine learning |
0.9 | 2 | 2021 | CrypTen: Secure Multi-Party Computation Meets Machine Learning · NeurIPS 2021 Certified Data Removal from Machine Learning Models · ICML 2020 |
Natural language and speech › Speech recognition and synthesis › automatic speech recognition
end-to-end speech recognition |
0.6 | 2 | 2019 | A fully differentiable beam search decoder · ICML 2019 Deep Speech 2 : End-to-End Speech Recognition in English and Mandarin · ICML 2016 |
Machine learning › Transfer learning and domain adaptation
domain generalization |
0.6 | 1 | 2022 | Gradient Matching for Domain Generalization · ICLR 2022 |
Machine learning › Efficient and distributed learning
gradient matching |
0.6 | 1 | 2022 | Gradient Matching for Domain Generalization · ICLR 2022 |
Privacy and data protection › information leakage
information leakage quantification |
0.6 | 1 | 2022 | Measuring Data Leakage in Machine-Learning Models with Fisher Information (Extended Abstract) · IJCAI 2022 |
Security and privacy of machine learning
membership inference |
0.6 | 1 | 2022 | Measuring Data Leakage in Machine-Learning Models with Fisher Information (Extended Abstract) · IJCAI 2022 |
Privacy and data protection
privacy-preserving data analysis |
0.6 | 1 | 2022 | Measuring Data Leakage in Machine-Learning Models with Fisher Information (Extended Abstract) · IJCAI 2022 |
Machine learning › Trustworthy machine learning
robustness |
0.5 | 1 | 2021 | Fixes That Fail: Self-Defeating Improvements in Machine-Learning Systems · NeurIPS 2021 |
Cryptographic protocols and secure computation
secure multiparty computation |
0.5 | 1 | 2021 | CrypTen: Secure Multi-Party Computation Meets Machine Learning · NeurIPS 2021 |
Machine learning › Representation and self-supervised learning › word representation
word embedding |
0.4 | 1 | 2020 | Word-Level Speech Recognition With a Letter to Word Encoder · ICML 2020 |
Security and privacy of machine learning
machine unlearning |
0.4 | 1 | 2020 | Certified Data Removal from Machine Learning Models · ICML 2020 |
Natural language and speech › Language models and text generation › decoding › decoding strategy
beam search |
0.4 | 1 | 2019 | A fully differentiable beam search decoder · ICML 2019 |
Machine learning › Deep learning architectures and training › neural network training
end-to-end deep learning |
0.2 | 1 | 2016 | Deep Speech 2 : End-to-End Speech Recognition in English and Mandarin · ICML 2016 |
GPUs and heterogeneous computing
deep learning on GPUs |
0.2 | 1 | 2016 | Persistent RNNs: Stashing Recurrent Weights On-Chip · ICML 2016 |
GPUs and heterogeneous computing
GPU computing |
0.2 | 1 | 2016 | Persistent RNNs: Stashing Recurrent Weights On-Chip · ICML 2016 |
Machine learning › Efficient and distributed learning
distributed training |
0.1 | 1 | 2021 | CrypTen: Secure Multi-Party Computation Meets Machine Learning · NeurIPS 2021 |
Robotics › Autonomous driving › perception
perception systems |
0.1 | 1 | 2021 | Fixes That Fail: Self-Defeating Improvements in Machine-Learning Systems · NeurIPS 2021 |
Machine learning › Deep learning architectures and training
convolutional neural network |
0.1 | 1 | 2019 | Ambulatory Atrial Fibrillation Monitoring Using Wearable Photoplethysmography with Deep Learning · KDD 2019 |
High-performance computing
performance optimization at scale |
0.1 | 1 | 2016 | Deep Speech 2 : End-to-End Speech Recognition in English and Mandarin · ICML 2016 |
Methods — techniques the papers use, named apart from their topics
tensor computation · 1.0automatic differentiation · 1.0certified removal mechanism · 0.9attention · 0.8gradient matching · 0.6fisher information · 0.6cramer-rao bound · 0.6secure multiparty computation · 0.5secure multi-party computation · 0.5retraining analysis · 0.5error decomposition · 0.5linear classifier · 0.4encoder-decoder · 0.4connectionist temporal classification · 0.4deep learning · 0.4convolutional neural network · 0.4recurrent neural network · 0.2persistent kernel · 0.2
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2022 | Data Appraisal Without Data SharingabstractOne of the most effective approaches to improving the performance of a machine learning model is to procure additional training data. A model owner seeking relevant training data from a data owner needs to appraise the data before acquiring it. However, without a formal agreement, the data owner does not want to share data. The resulting Catch-22 prevents efficient data markets from forming. This paper proposes adding a data appraisal stage that requires no data sharing between data owners and model owners. Specifically, we use multi-party computation to implement an appraisal function computed on private data. The appraised value serves as a guide to facilitate data selection and transaction. We propose an efficient data appraisal method based on forward influence functions that approximates data value through its first-order loss reduction on the current model. The method requires no additional hyper-parameters or re-training. We show that in private, forward influence functions provide an appealing trade-off between high quality appraisal and required computation, in spite of label noise, class imbalance, and missing data. Our work seeks to inspire an open market that incentivizes efficient, equitable exchange of domain-specific training data. Xinlei Xu, Awni Y. Hannun, Laurens van der Maaten |
AISTATS | 2 |
| 2022 | Parallel Composition of Weighted Finite-State TransducersabstractFinite-state transducers (FSTs) are frequently used in speech recognition. Transducer composition is an essential operation for combining different sources of information at different granularities. However, composition is also one of the more computationally expensive operations. Due to the heterogeneous structure of FSTs, parallel algorithms for composition are suboptimal in efficiency, generality, or both. We propose an algorithm for parallel composition and implement it on graphics processing units. We benchmark our parallel algorithm on the composition of random graphs and the composition of graphs commonly used in speech recognition. The parallel composition scales better with the size of the input graphs and for large graphs can be as much as 10 to 30 times faster than a sequential CPU algorithm. Shubho Sengupta, Vineel Pratap, Awni Y. Hannun |
ICASSP | 3 |
| 2022 | Gradient Matching for Domain Generalization
Yuge Shi, Jeffrey Seely, Philip Torr 0001, N. Siddharth 0001, Awni Y. Hannun, Nicolas Usunier, Gabriel Synnaeve |
ICLR | 5 |
| 2022 | Flashlight: Enabling Innovation in Tools for Machine LearningabstractAs the computational requirements for machine learning systems and the size and complexity of machine learning frameworks increases, essential framework innovation has become challenging. While computational needs have driven recent compiler, networking, and hardware advancements, utilization of those advancements by machine learning tools is occurring at a slower pace. This is in part due to the difficulties involved in prototyping new computational paradigms with existing frameworks. Large frameworks prioritize machine learning researchers and practitioners as end users and pay comparatively little attention to systems researchers who can push frameworks forward — we argue that both are equally important stakeholders. We introduce Flashlight, an open-source library built to spur innovation in machine learning tools and systems by prioritizing open, modular, customizable internals and state-of-the-art, research-ready models and training setups across a variety of domains. Flashlight allows systems researchers to rapidly prototype and experiment with novel ideas in machine learning computation and has low overhead, competing with and often outperforming other popular machine learning frameworks. We see Flashlight as a tool enabling research that can benefit widely used libraries downstream and bring machine learning and systems researchers closer together. Jacob Kahn, Vineel Pratap, Tatiana Likhomanenko, Qiantong Xu, Awni Y. Hannun, Jeff Cai, Paden Tomasello, Ann Lee 0001, Edouard Grave, Gilad Avidov, Benoit Steiner, Vitaliy Liptchinsky, Gabriel Synnaeve, Ronan Collobert |
ICML | 5 |
| 2022 | Measuring Data Leakage in Machine-Learning Models with Fisher Information (Extended Abstract)abstractMachine-learning models contain information about the data they were trained on. This information leaks either through the model itself or through predictions made by the model. Consequently, when the training data contains sensitive attributes, assessing the amount of information leakage is paramount. We propose a method to quantify this leakage using the Fisher information of the model about the data. Unlike the worst-case a priori guarantees of differential privacy, Fisher information loss measures leakage with respect to specific examples, attributes, or sub-populations within the dataset. We motivate Fisher information loss through the Cramer-Rao bound and delineate the implied threat model. We provide efficient methods to compute Fisher information loss for output-perturbed generalized linear models. Finally, we empirically validate Fisher information loss as a useful measure of information leakage. Awni Y. Hannun, Chuan Guo 0001, Laurens van der Maaten |
IJCAI | 1 |
| 2021 | CrypTen: Secure Multi-Party Computation Meets Machine LearningabstractSecure multi-party computation (MPC) allows parties to perform computations on data while keeping that data private. This capability has great potential for machine-learning applications: it facilitates training of machine-learning models on private data sets owned by different parties, evaluation of one party's private model using another party's private data, etc. Although a range of studies implement machine-learning models via secure MPC, such implementations are not yet mainstream. Adoption of secure MPC is hampered by the absence of flexible software frameworks that `"speak the language" of machine-learning researchers and engineers. To foster adoption of secure MPC in machine learning, we present CrypTen: a software framework that exposes popular secure MPC primitives via abstractions that are common in modern machine-learning frameworks, such as tensor computations, automatic differentiation, and modular neural networks. This paper describes the design of CrypTen and measure its performance on state-of-the-art models for text classification, speech recognition, and image classification. Our benchmarks show that CrypTen's GPU support and high-performance communication between (an arbitrary number of) parties allows it to perform efficient private evaluation of modern machine-learning models under a semi-honest threat model. For example, two parties using CrypTen can securely predict phonemes in speech recordings using Wav2Letter faster than real-time. We hope that CrypTen will spur adoption of secure MPC in the machine-learning community. Brian Knott, Shobha Venkataraman, Awni Y. Hannun, Shubho Sengupta, Mark Ibrahim, Laurens van der Maaten |
NeurIPS | 3 |
| 2021 | Fixes That Fail: Self-Defeating Improvements in Machine-Learning SystemsabstractMachine-learning systems such as self-driving cars or virtual assistants are composed of a large number of machine-learning models that recognize image content, transcribe speech, analyze natural language, infer preferences, rank options, etc. Models in these systems are often developed and trained independently, which raises an obvious concern: Can improving a machine-learning model make the overall system worse? We answer this question affirmatively by showing that improving a model can deteriorate the performance of downstream models, even after those downstream models are retrained. Such self-defeating improvements are the result of entanglement between the models in the system. We perform an error decomposition of systems with multiple machine-learning models, which sheds light on the types of errors that can lead to self-defeating improvements. We also present the results of experiments which show that self-defeating improvements emerge in a realistic stereo-based detection system for cars and pedestrians. Ruihan Wu, Chuan Guo 0001, Awni Y. Hannun, Laurens van der Maaten |
NeurIPS | 3 |
| 2021 | Semi-Supervised end-to-end Speech Recognition via Local Prior MatchingabstractFor sequence transduction tasks like speech recognition, a strong structured prior model encodes rich information about the target space, implicitly ruling out invalid sequences by assigning them low probability. In this work, we propose local prior matching (LPM), a semi-supervised objective that distills knowledge from a strong prior (e.g. a language model) to provide learning signal to an end-to-end model trained on unlabeled speech. We demonstrate that LPM is simple to implement and superior to existing knowledge distillation techniques under comparable settings. Starting from a baseline trained on 100 hours of labeled speech, with an additional 360 hours of unlabeled data, LPM recovers 54%/82% and 73%/91% of the word error rate on clean and noisy test sets with/without language model rescoring relative to a fully supervised model on the same data. Wei-Ning Hsu, Ann Lee 0001, Gabriel Synnaeve, Awni Y. Hannun |
SLT | 4 |
| 2021 | Measuring data leakage in machine-learning models with Fisher informationabstractMachine-learning models contain information about the data they were trained on. This information leaks either through the model itself or through predictions made by the model. Consequently, when the training data contains sensitive attributes, assessing the amount of information leakage is paramount. We propose a method to quantify this leakage using the Fisher information of the model about the data. Unlike the worst-case a priori guarantees of differential privacy, Fisher information loss measures leakage with respect to specific examples, attributes, or sub-populations within the dataset. We motivate Fisher information loss through the Cram\’{e}r-Rao bound and delineate the implied threat model. We provide efficient methods to compute Fisher information loss for output-perturbed generalized linear models. Finally, we empirically validate Fisher information loss as a useful measure of information leakage. Awni Y. Hannun, Chuan Guo 0001, Laurens van der Maaten |
UAI | 1 |
| 2020 | Self-Training for End-to-End Speech RecognitionabstractWe revisit self-training in the context of end-to-end speech recognition. We demonstrate that training with pseudo-labels can substantially improve the accuracy of a baseline model. Key to our approach are a strong baseline acoustic and language model used to generate the pseudo-labels, filtering mechanisms tailored to common errors from sequence-to-sequence models, and a novel ensemble approach to increase pseudo-label diversity. Experiments on the LibriSpeech corpus show that with an ensemble of four models and label filtering, self-training yields a 33.9% relative improvement in WER compared with a baseline trained on 100 hours of labelled data in the noisy speech setting. In the clean speech setting, self-training recovers 59.3% of the gap between the baseline and an oracle model, which is at least 93.8% relatively higher than what previous approaches can achieve. Jacob Kahn, Ann Lee 0001, Awni Y. Hannun |
ICASSP | 3 |
| 2020 | Word-Level Speech Recognition With a Letter to Word EncoderabstractWe propose a direct-to-word sequence model which uses a word network to learn word embeddings from letters. The word network can be integrated seamlessly with arbitrary sequence models including Connectionist Temporal Classification and encoder-decoder models with attention. We show our direct-to-word model can achieve word error rate gains over sub-word level models for speech recognition. We also show that our direct-to-word approach retains the ability to predict words not seen at training time without any retraining. Finally, we demonstrate that a word-level model can use a larger stride than a sub-word level model while maintaining accuracy. This makes the model more efficient both for training and inference. Ronan Collobert, Awni Y. Hannun, Gabriel Synnaeve |
ICML | 2 |
| 2020 | Certified Data Removal from Machine Learning ModelsabstractGood data stewardship requires removal of data at the request of the data’s owner. This raises the question if and how a trained machine-learning model, which implicitly stores information about its training data, should be affected by such a removal request. Is it possible to “remove” data from a machine-learning model? We study this problem by defining certified removal: a very strong theoretical guarantee that a model from which data is removed cannot be distinguished from a model that never observed the data to begin with. We develop a certified-removal mechanism for linear classifiers and empirically study learning settings in which this mechanism is practical. Chuan Guo 0001, Tom Goldstein, Awni Y. Hannun, Laurens van der Maaten |
ICML | 3 |
| 2020 | Massively Multilingual ASR: 50 Languages, 1 Model, 1 Billion ParametersabstractWe study training a single acoustic model for multiple languages with the aim of improving automatic speech recognition (ASR) performance on low-resource languages, and over-all simplifying deployment of ASR systems that support diverse languages. We perform an extensive benchmark on 51 languages, with varying amount of training data by language(from 100 hours to 1100 hours). We compare three variants of multilingual training from a single joint model without knowing the input language, to using this information, to multiple heads (one per language cluster). We show that multilingual training of ASR models on several languages can improve recognition performance, in particular, on low resource languages. We see 20.9%, 23% and 28.8% average WER relative reduction compared to monolingual baselines on joint model, joint model with language input and multi head model respectively. To our knowledge, this is the first work studying multilingual ASR at massive scale, with more than 50 languages and more than 16,000 hours of audio across them. Vineel Pratap, Anuroop Sriram, Paden Tomasello, Awni Y. Hannun, Vitaliy Liptchinsky, Gabriel Synnaeve, Ronan Collobert |
INTERSPEECH | 4 |
| 2020 | Scaling Up Online Speech Recognition Using ConvNetsabstractWe design an online end-to-end speech recognition system based on Time-Depth Separable (TDS) convolutions and Connectionist Temporal Classification (CTC). We improve the core TDS architecture in order to limit the future context and hence reduce latency while maintaining accuracy. The system has almost three times the throughput of a well tuned hybrid ASR baseline while also having lower latency and a better word error rate. Also important to the efficiency of the recognizer is our highly optimized beam search decoder. To show the impact of our design choices, we analyze throughput, latency, accuracy, and discuss how these metrics can be tuned based on the user requirements. Vineel Pratap, Qiantong Xu, Jacob Kahn, Gilad Avidov, Tatiana Likhomanenko, Awni Y. Hannun, Vitaliy Liptchinsky, Gabriel Synnaeve, Ronan Collobert |
INTERSPEECH | 6 |
| 2020 | Iterative Pseudo-Labeling for Speech RecognitionabstractPseudo-labeling has recently shown promise in end-to-end automatic speech recognition (ASR).We study Iterative Pseudo-Labeling (IPL), a semi-supervised algorithm which efficiently performs multiple iterations of pseudo-labeling on unlabeled data as the acoustic model evolves.In particular, IPL fine tunes an existing model at each iteration using both labeled data and a subset of unlabeled data.We study the main components of IPL: decoding with a language model and data augmentation.We then demonstrate the effectiveness of IPL by achieving state-of-the-art word-error rate on the LIBRISPEECH test sets in both standard and low-resource setting.We also study the effect of language models trained on different corpora to show IPL can effectively utilize additional text.Finally, we release a new large in-domain text corpus which does not overlap with the LIBRISPEECH training transcriptions to foster research in low-resource, semi-supervised ASR. Qiantong Xu, Tatiana Likhomanenko, Jacob Kahn, Awni Y. Hannun, Gabriel Synnaeve, Ronan Collobert |
INTERSPEECH | 4 |
| 2019 | Lead2Gold: Towards Exploiting the Full Potential of Noisy Transcriptions for Speech RecognitionabstractThe transcriptions used to train an Automatic Speech Recognition (ASR) system may contain errors. Usually, either a quality control stage discards transcriptions with too many errors, or the noisy transcriptions are used as is. We introduce Lead2Gold, a method to train an ASR system that exploits the full potential of noisy transcriptions. Based on a noise model of transcription errors, Lead2Gold searches for better transcriptions of the training data with a beam search that takes this noise model into account. The beam search is differentiable and does not require a forced alignment step, thus the whole system is trained end-to-end. Lead2Gold can be viewed as a new loss function that can be used on top of any sequence-to-sequence deep neural network. We conduct proof-of-concept experiments on noisy transcriptions generated from letter corruptions with different noise levels. We show that Lead2Gold obtains a better ASR accuracy than a competitive baseline which does not account for the (artificially-introduced) transcription noise. Adrien Dufraux, Emmanuel Vincent 0001, Awni Y. Hannun, Armelle Brun, Matthijs Douze |
ASRU | 3 |
| 2019 | Wav2Letter++: A Fast Open-source Speech Recognition SystemabstractThis paper introduces wav2letter++, a fast open-source deep learning speech recognition framework. wav2letter++ is written entirely in C++, and uses the ArrayFire tensor library for maximum efficiency. We explain the architecture and design of the wav2letter++ system and compare it to other major open-source speech recognition systems. In some cases wav2letter++ is more than 2× faster than other optimized frameworks for training end-to-end neural networks for speech recognition. We also show that wav2letter++ training times scale linearly to 64 GPUs, the most we tested, for models with 100 million parameters. High-performance frameworks enable fast iteration, which is often a crucial factor in successful research and model tuning on new datasets and tasks. Vineel Pratap, Awni Y. Hannun, Qiantong Xu, Jeff Cai, Jacob Kahn, Gabriel Synnaeve, Vitaliy Liptchinsky, Ronan Collobert |
ICASSP | 2 |
| 2019 | A fully differentiable beam search decoderabstractWe introduce a new beam search decoder that is fully differentiable, making it possible to optimize at training time through the inference procedure. Our decoder allows us to combine models which operate at different granularities (e.g. acoustic and language models). It can be used when target sequences are not aligned to input sequences by considering all possible alignments between the two. We demonstrate our approach scales by applying it to speech recognition, jointly training acoustic and word-level language models. The system is end-to-end, with gradients flowing through the whole architecture from the word-level transcriptions. Recent research efforts have shown that deep neural networks with attention-based mechanisms can successfully train an acoustic model from the final transcription, while implicitly learning a language model. Instead, we show that it is possible to discriminatively train an acoustic model jointly with an explicit and possibly pre-trained language model. Ronan Collobert, Awni Y. Hannun, Gabriel Synnaeve |
ICML | 2 |
| 2019 | Sequence-to-Sequence Speech Recognition with Time-Depth Separable ConvolutionsabstractWe propose a fully convolutional sequence-to-sequence encoder architecture with a simple and efficient decoder. Our model improves WER on LibriSpeech while being an order of magnitude more efficient than a strong RNN baseline. Key to our approach is a time-depth separable convolution block which dramatically reduces the number of parameters in the model while keeping the receptive field large. We also give a stable and efficient beam search inference procedure which allows us to effectively integrate a language model. Coupled with a convolutional language model, our time-depth separable convolution architecture improves by more than 22% relative WER over the best previously reported sequence-to-sequence results on the noisy LibriSpeech test set. Awni Y. Hannun, Ann Lee 0001, Qiantong Xu, Ronan Collobert |
INTERSPEECH | 1 |
| 2019 | Ambulatory Atrial Fibrillation Monitoring Using Wearable Photoplethysmography with Deep LearningabstractWe develop an algorithm that accurately detects Atrial Fibrillation (AF) episodes from photoplethysmograms (PPG) recorded in ambulatory free-living conditions. We collect and annotate a dataset containing more than 4000 hours of PPG recorded from a wrist-worn device. Using a 50-layer convolutional neural network, we achieve a test AUC of 95% in presence of motion artifacts inherent to PPG signals. Such continuous and accurate detection of AF has the potential to transform consumer wearable devices into clinically useful medical monitoring tools. Yichen Shen 0003, Maxime Voisin, Alireza Aliamiri, Anand Avati, Awni Y. Hannun, Andrew Y. Ng |
KDD | 5 |
| 2017 | Building DNN acoustic models for large vocabulary speech recognition
Andrew L. Maas, Peng Qi 0003, Ziang Xie, Awni Y. Hannun, Christopher T. Lengerich, Daniel Jurafsky, Andrew Y. Ng |
Comput. Speech Lang. | 4 |
| 2016 | Deep Speech 2 : End-to-End Speech Recognition in English and MandarinabstractWe show that an end-to-end deep learning approach can be used to recognize either English or Mandarin Chinese speech–two vastly different languages. Because it replaces entire pipelines of hand-engineered components with neural networks, end-to-end learning allows us to handle a diverse variety of speech including noisy environments, accents and different languages. Key to our approach is our application of HPC techniques, enabling experiments that previously took weeks to now run in days. This allows us to iterate more quickly to identify superior architectures and algorithms. As a result, in several cases, our system is competitive with the transcription of human workers when benchmarked on standard datasets. Finally, using a technique called Batch Dispatch with GPUs in the data center, we show that our system can be inexpensively deployed in an online setting, delivering low latency when serving users at scale. Dario Amodei, Sundaram Ananthanarayanan, Rishita Anubhai, Jingliang Bai, Eric Battenberg, Carl Case, Jared Casper, Bryan Catanzaro, Jingdong Chen, Mike Chrzanowski, Adam Coates 0002, Gregory Frederick Diamos, Erich Elsen, Jesse H. Engel, Linxi Fan, Christopher Fougner, Awni Y. Hannun, Billy Jun, Tony Han, Patrick LeGresley, Xiangang Li, Libby Lin, Sharan Narang, Andrew Y. Ng, Sherjil Ozair, Ryan Prenger, Sheng Qian, Jonathan Raiman, Sanjeev Satheesh, David Seetapun, Shubho Sengupta, Chong Wang 0002, Zhiqian Wang, Dani Yogatama, Zhenyao Zhu |
ICML | 17 |
| 2016 | Persistent RNNs: Stashing Recurrent Weights On-ChipabstractThis paper introduces a new technique for mapping Deep Recurrent Neural Networks (RNN) efficiently onto GPUs. We show how it is possi- ble to achieve substantially higher computational throughput at low mini-batch sizes than direct implementations of RNNs based on matrix multiplications. The key to our approach is the use of persistent computational kernels that exploit the GPU’s inverted memory hierarchy to reuse network weights over multiple timesteps. Our initial implementation sustains 2.8 TFLOP/s at a mini-batch size of 4 on an NVIDIA TitanX GPU. This provides a 16x reduction in activation memory footprint, enables model training with 12x more parameters on the same hardware, allows us to strongly scale RNN training to 128 GPUs, and allows us to efficiently explore end-to-end speech recognition models with over 100 layers. Gregory Frederick Diamos, Shubho Sengupta, Bryan Catanzaro, Mike Chrzanowski, Adam Coates 0002, Erich Elsen, Jesse H. Engel, Awni Y. Hannun, Sanjeev Satheesh |
ICML | 8 |
| 2016 | Learning Multiscale Features Directly from WaveformsabstractDeep learning has dramatically improved the performance of speech recognition systems through learning hierarchies of features optimized for the task at hand. However, true end-to-end learning, where features are learned directly from waveforms, has only recently reached the performance of hand-tailored representations based on the Fourier transform. In this paper, we detail an approach to use convolutional filters to push past the inherent tradeoff of temporal and frequency resolution that exists for spectral representations. At increased computational cost, we show that increasing temporal resolution via reduced stride and increasing frequency resolution via additional filters delivers significant performance improvements. Further, we find more efficient representations by simultaneously learning at multiple scales, leading to an overall decrease in word error rate on a difficult internal speech test set by 20.7% relative to networks with the same number of parameters trained on spectrograms. Zhenyao Zhu, Jesse H. Engel, Awni Y. Hannun |
INTERSPEECH | 3 |