US 12,260,616 B1
Multi-task machine learning model for event detection
Narendran Rajan, San Diego, CA (US); Yan Wang, Mercer Island, WA (US); Phil Ammirato, Campbell, CA (US); Kevin Lai, Redmond, WA (US); Evan Welbourne, Seattle, WA (US); and Nathan Hurst, Seattle, WA (US)
Assigned to Samsara Inc., San Francisco, CA (US)
Filed by Samsara Inc., San Francisco, CA (US)
Filed on Jun. 14, 2024, as Appl. No. 18/743,824.
Int. Cl. G06K 9/00 (2022.01); G06V 10/764 (2022.01)
CPC G06V 10/764 (2022.01) 20 Claims
OG exemplary drawing
 
1. A computer-implemented method for machine learning model operation, the computer-implemented method comprising, by one or more processors executing program instructions:
providing a first training dataset comprising a plurality of images and associated object detection labels;
providing a second training dataset comprising a plurality of images and associated classification labels, wherein the second training dataset does not include object detection labels;
providing a machine learning model comprising a model backbone, an object detection task head, and a classification task head;
training the machine learning model, wherein training the machine learning model comprises:
training the object detection task head using the first training dataset; and
training the classification task head using the second training dataset; and
deploying the trained machine learning model for use, wherein the trained machine learning model includes the trained model backbone and the trained object detection task head, and wherein the trained machine learning model does not include the trained classification task head.