[發明專利]基于存儲器的量選擇神經網絡

在審

• 申請號: 202210145285.2
• 申請日: 2022-02-17
• 公開（公告）號: CN115018040A
• 公開（公告）日: 2022-09-06
• 發明（設計）人: 楊棟;何宇帆;H·R·羅斯;趙燦;徐大光
• 申請（專利權）人: 輝達公司
• 主分類號: G06N3/04
• 分類號: G06N3/04;G06T7/10
• 代理公司: 北京市磐華律師事務所 11336
• 代理人: 高偉

1.一種處理器，包括：一個或更多個電路，用于使一個或更多個神經網絡至少部分地基于將由所述一個或更多個神經網絡使用的存儲器的量而從多個神經網絡中被選擇。

2.如權利要求1所述的處理器，其中所述一個或更多個電路進一步用于執行搜索以從所述多個神經網絡中選擇滿足存儲器約束的所述一個或更多個神經網絡。

3.如權利要求1所述的處理器，其中所述一個或更多個電路進一步用于使一個或更多個附加神經網絡從所述多個神經網絡或從第二多個神經網絡中選擇，其中所述一個或更多個附加神經網絡與所述一個或更多個神經網絡不同并且滿足第二存儲器約束，所述第二存儲器約束不同于由所述一個或更多個神經網絡滿足的第一存儲器。

4.如權利要求1所述的處理器，其中所述一個或更多個電路進一步用于執行搜索，以根據至少部分地基于將由所述一個或更多個神經網絡使用的存儲器的量確定的一組一個或更多個搜索參數而從所述多個神經網絡中選擇所述一個或更多個神經網絡。

5.如權利要求1所述的處理器，其中與包括所述多個神經網絡的搜索空間的一個或更多個候選特征節點相關聯的操作的最大存儲器使用量的百分比小于或等于所述存儲器的量。

6.如權利要求1所述的處理器，其中所述一個或更多個電路進一步通過執行拓撲搜索空間和單元搜索空間的聯合兩級搜索來識別用于基于圖像的任務的所述一個或更多個神經網絡，使所述一個或更多個神經網絡被選擇。

7.一種系統，包括：

一個或更多個處理器，用于使一個或更多個神經網絡至少部分地基于將由所述一個或更多個神經網絡使用的存儲器的量而從多個神經網絡中被選擇；以及

一個或更多個存儲器，用于存儲對應于所述一個或更多個神經網絡的參數。

8.如權利要求7所述的系統，其中所述一個或更多個處理器進一步用于執行第一搜索以從所述多個神經網絡中選擇滿足第一存儲器約束的所述一個或更多個神經網絡。

9.如權利要求7所述的系統，其中所述一個或更多個處理器進一步用于至少部分地基于所述一個或更多個神經網絡的第一層和第二層之間的多個候選連接模式中的每一個的概率，從所述多個候選連接模式中選擇連接模式。

10.如權利要求7所述的系統，其中所述一個或更多個處理器進一步用于從所述一個或更多個神經網絡的一層或更多層的一組候選特征節點中選擇特征節點，其中所述一組候選特征節點包括不同圖像尺度的特征節點，所述特征節點包括連接到前一層中的特征節點的多個候選邊。

11.如權利要求7所述的系統，其中所述一個或更多個處理器進一步通過執行拓撲搜索空間和單元搜索空間的聯合兩級搜索來識別用于基于圖像的任務的所述一個或更多個神經網絡，使所述一個或更多個神經網絡被選擇。

12.如權利要求7所述的系統，其中所述一個或更多個神經網絡用于執行圖像分割任務。

13.如權利要求7所述的系統，其中所述一個或更多個處理器進一步通過執行拓撲搜索空間和單元搜索空間的搜索來使所述一個或更多個神經網絡被選擇，所述拓撲搜索空間包括連接多個層的候選特征節點的多個候選邊，所述單元搜索空間包括多個候選操作。

14.如權利要求7所述的系統，其中所述一個或更多個處理器進一步用于至少部分地基于所述一個或更多個神經網絡的層之間的多個候選連接模式的概率，從所述多個候選連接模式中選擇連接模式。

15.一種數據中心，包括：一個或更多個處理器，用于使一個或更多個神經網絡至少部分地基于將由所述一個或更多個神經網絡使用的存儲器的量而從多個神經網絡中被選擇來執行醫學圖像分割任務。