Planter: rapid prototyping of in-network machine learning inference

Journal article

In-network machine learning inference provides high throughput and low latency. It is ideally located within the network, power efficient, and improves applications' performance. Despite its advantages, the bar to in-network machine learning research is high, requiring significant expertise in programmable data planes, in addition to knowledge of machine learning and the application area. Existing solutions are mostly one-time efforts, hard to reproduce, change, or port across platforms. In this paper, we present Planter: a modular and efficient open-source framework for rapid prototyping of in-network machine learning models across a range of platforms and pipeline architectures. By identifying general mapping methodologies for machine learning algorithms, Planter introduces new machine learning mappings and improves existing ones. It provides users with several example use cases and supports different datasets, and was already extended by users to new fields and applications. Our evaluation shows that Planter improves machine learning performance compared with previous model-tailored works, while significantly reducing resource consumption and co-existing with network functionality. Planter-supported algorithms run at line rate on unmodified commodity hardware, providing billions of inference decisions per second.

Authors: Zheng, C; Zang, M; Hong, X; Perreault, L; Bensoussane, R

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: Association for Computing Machinery
• Journal: ACM SIGCOMM Computer Communication Review
• Volume: 54
• Issue: 1
• Pages: 2 - 21
• Publication date: 2024-08-06
• Acceptance date: 2024-04-02
• DOI: 10.1145/3687230.3687232
• ISSN: 0146-4833
• Language: English
• Keywords: in-network computing; programmable switches; modular framework; machine learning compilers; machine learning; P4; dimension reduction