IP Library Granted Patent US 10,755,147
Granted Patent B2
US 10,755,147 · App. 16/046,084 · Granted Aug 25, 2020

Classification and localization based on annotation information

Inventors: Qian Zhao (Dublin, CA); Min Zhang (San Ramon, CA); Gopal Avinash (San Ramon, CA)
Assignee: General Electric Company
G06K9/6262G06K9/6269G06K9/6278G06K2209/05
View Patent ↗
Loading inventors, assignments & file history…
Monitor This Case
Get email alerts when status or documents change.
Order Certified Copies
Most orders are placed with the USPTO same day — all within 24 business hours.
Order via The Patent Place →
Pre-filled with this patent's details
Quick Facts
Patent No.
US 10,755,147
App. No.
16/046,084
Granted
Aug 25, 2020
Kind
B2
Abstract

Systems and techniques for classification and localization based on annotation information are presented. In one example, a system trains a convolutional neural network based on training data and a plurality of images. The training data is associated with a plurality of patients from at least one imaging device. The plurality of images is associated with a plurality of masks from a plurality of objects. The convolutional neural network comprises a decoder consisting of at least one up-sampling layer and at least one convolutional layer. The system also generates a loss function based on the plurality of masks, where the loss function is iteratively back propagated to tune parameters of the convolutional neural network. The system also predicts a classification label for an input image based on the convolutional neural network.

Claims (37)

1. A machine learning system that employs masks and decoders for improved performance, comprising:

a memory that stores computer executable components;

a processor that executes computer executable components stored in the memory, wherein the computer executable components comprise:

a training component that trains a convolutional neural network based on training data and a plurality of images, wherein the training data is associated with a plurality of patients from at least one imaging device, wherein the plurality of images is associated with a plurality of masks from a plurality of objects, and wherein the convolutional neural network comprises a decoder consisting of at least one up-sampling layer and at least one convolutional layer, wherein the decoder is a set of decoders, and wherein a number of decoders included in the set of decoders is determined during training of the convolutional neural network;

a loss function component that generates a loss function based on the plurality of masks, wherein the loss function is iteratively back propagated to tune parameters of the convolutional neural network;

a classification component that predicts a classification label for an input image based on the convolutional neural network; and

a visualization component that generates a multi-dimensional visualization associated with the classification label for the input image.

2. The machine learning system of claim 1 , wherein the convolutional neural network comprises a pretrained classifier network that outputs convolutional feature maps.

3. The machine learning system of claim 2 , wherein the convolutional neural network comprises a classification/localization network that outputs corresponding scoring maps based on the convolutional feature maps.

4. The machine learning system of claim 2 , wherein a size of a mask from the plurality of masks is matched with a size of a convolutional feature map from the convolutional feature maps.

5. The machine learning system of claim 2 , wherein a size of a mask from the plurality of masks is matched with a size of a convolutional feature map from the convolutional feature maps based on a max pooling process.

6. The machine learning system of claim 1 , wherein the loss function component employs the decoder to generate the localization map.

7. The machine learning system of claim 1 , wherein the loss function component generates the loss function based on a probability for a class associated with the plurality of masks.

8. A method that employs masks and decoders for improved performance, comprising using a processor operatively coupled to memory to execute computer executable components to perform the following acts:

receiving a plurality of images associated with a plurality of patients from at least one imaging device;

receiving a plurality of masks from a plurality of objects, wherein each image comprises at least one mask associating an object of interest with a corresponding class label;

training a convolutional neural network based on the plurality of images and the plurality of masks, wherein the convolutional neural network comprises a decoder consisting of at least one up-sampling layer and at least one convolutional layer, a pretrained classifier network that outputs convolutional feature maps, and a classification/localization network that outputs corresponding scoring maps, wherein the decoder is a set of decoders,;

determining a number of decoders in the set of decoders during the training of the convolutional neural network;

generating a loss function based on the plurality of masks;

iteratively back propagating the loss function to tune parameters of the convolutional neural network;

predicting a classification label for an input image based on the convolutional neural network; and

generating a multi-dimensional visualization associated with the classification label for the input image.

9. The method of claim 8 , further comprising matching a size of a mask from the plurality of masks with a size of a convolutional feature map from the convolutional feature maps.

10. The method of claim 8 , further comprising matching a size of a mask from the plurality of masks with a size of a convolutional feature map from the convolutional feature maps based on a max pooling process.

11. The method of claim 8 , wherein the generating the loss function comprises employing the decoder to generate the localization map.

12. The method of claim 8 , wherein the generating the loss function comprises generating the loss function based on a probability for a class associated with the plurality of masks.

13. A computer readable storage device comprising instructions that, in response to execution, cause a system comprising a processor to perform operations that employ masks and decoders for improved performance, comprising:

receiving a plurality of images associated with a plurality of patients from at least one imaging device;

receiving a plurality of masks from a plurality of objects, wherein each image comprises at least one mask associating an object of interest with a corresponding class label;

training a convolutional neural network based on the plurality of images and the plurality of masks, wherein the convolutional neural network comprises a decoder consisting of at least one up-sampling layer and at least one convolutional layer, a pretrained classifier network that outputs convolutional feature maps, and a classification/localization network that outputs corresponding scoring maps, wherein the decoder is a set of decoders;

determining a number of decoders in the set of decoders during the training of the convolutional neural network;

generating a loss function based on the plurality of masks;

iteratively back propagating the loss function to tune parameters of the convolutional neural network;

predicting a classification label for an input image based on the convolutional neural network; and

generating a multi-dimensional visualization associated with the classification label for the input image.

14. The computer readable storage device of claim 13 , wherein the generating the loss function comprises employing the set of decoders to generate the localization map.

15. The computer readable storage device of claim 13 , wherein the generating the loss function comprises generating the loss function based on a probability for a class associated with the plurality of masks.

Assignments (2)
NUNC PRO TUNC ASSIGNMENT Recorded May 8, 2025
From: GENERAL ELECTRIC COMPANY
To: GE PRECISION HEALTHCARE LLC
Reel/Frame 071225/0218 →
ASSIGNMENT OF ASSIGNOR'S INTEREST Recorded Jul 26, 2018
From: ZHAO, QIAN; ZHANG, MIN; AVINASH, GOPAL
To: GENERAL ELECTRIC COMPANY
Reel/Frame 046468/0412 →
Continuity (2)
Provisional Application 62693772 · Jul 3, 2018
Related Publication 20200012895A1 · Jan 9, 2020