CrossNAS: A Cross-Layer Neural Architecture Search Framework for PIM Systems
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909673929572352 |
|---|---|
| author | Amin, Md Hasibul Mohammadi, Mohammadreza Bakos, Jason D. Zand, Ramtin |
| author_facet | Amin, Md Hasibul Mohammadi, Mohammadreza Bakos, Jason D. Zand, Ramtin |
| contents | In this paper, we propose the CrossNAS framework, an automated approach for exploring a vast, multidimensional search space that spans various design abstraction layers-circuits, architecture, and systems-to optimize the deployment of machine learning workloads on analog processing-in-memory (PIM) systems. CrossNAS leverages the single-path one-shot weight-sharing strategy combined with the evolutionary search for the first time in the context of PIM system mapping and optimization. CrossNAS sets a new benchmark for PIM neural architecture search (NAS), outperforming previous methods in both accuracy and energy efficiency while maintaining comparable or shorter search times. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2505_22868 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | CrossNAS: A Cross-Layer Neural Architecture Search Framework for PIM Systems Amin, Md Hasibul Mohammadi, Mohammadreza Bakos, Jason D. Zand, Ramtin Emerging Technologies Hardware Architecture Machine Learning In this paper, we propose the CrossNAS framework, an automated approach for exploring a vast, multidimensional search space that spans various design abstraction layers-circuits, architecture, and systems-to optimize the deployment of machine learning workloads on analog processing-in-memory (PIM) systems. CrossNAS leverages the single-path one-shot weight-sharing strategy combined with the evolutionary search for the first time in the context of PIM system mapping and optimization. CrossNAS sets a new benchmark for PIM neural architecture search (NAS), outperforming previous methods in both accuracy and energy efficiency while maintaining comparable or shorter search times. |
| title | CrossNAS: A Cross-Layer Neural Architecture Search Framework for PIM Systems |
| topic | Emerging Technologies Hardware Architecture Machine Learning |
| url | https://arxiv.org/abs/2505.22868 |