Fastai: A Layered API for Deep Learning

被引:594
作者
Howard, Jeremy [1 ,2 ]
Gugger, Sylvain [1 ]
机构
[1] Fastai, San Francisco, CA 94117 USA
[2] Univ San Francisco, Data Sci Inst, San Francisco, CA 94117 USA
关键词
deep learning; data processing pipelines;
D O I
10.3390/info11020108
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
fastai is a deep learning library which provides practitioners with high-level components that can quickly and easily provide state-of-the-art results in standard deep learning domains, and provides researchers with low-level components that can be mixed and matched to build new approaches. It aims to do both things without substantial compromises in ease of use, flexibility, or performance. This is possible thanks to a carefully layered architecture, which expresses common underlying patterns of many deep learning and data processing techniques in terms of decoupled abstractions. These abstractions can be expressed concisely and clearly by leveraging the dynamism of the underlying Python language and the flexibility of the PyTorch library. fastai includes: a new type dispatch system for Python along with a semantic type hierarchy for tensors; a GPU-optimized computer vision library which can be extended in pure Python; an optimizer which refactors out the common functionality of modern optimizers into two basic pieces, allowing optimization algorithms to be implemented in 4-5 lines of code; a novel 2-way callback system that can access any part of the data, model, or optimizer and change it at any point during training; a new data block API; and much more. We used this library to successfully create a complete deep learning course, which we were able to write more quickly than using previous approaches, and the code was more clear. The library is already in wide use in research, industry, and teaching.
引用
收藏
页数:26
相关论文
共 62 条
[1]  
Abadi M., 2015, P 12 USENIX S OPERAT
[2]  
[Anonymous], 2018, ARXIV180106146
[3]  
[Anonymous], 2019, SELF REGULARIZED NON
[4]  
[Anonymous], 2010, MNIST handwritten digit database
[5]  
[Anonymous], 2015, ARTIFICIAL NEURAL NE
[6]  
[Anonymous], ICML
[7]  
[Anonymous], 2016, ARXIV160406737
[8]  
[Anonymous], SWIFTAI
[9]  
[Anonymous], 2018, 2018 INT C INT SYST
[10]  
[Anonymous], P IEEE C COMP VIS PA