Out-of-distribution detection of input instances to the model
By using secondary models trained on the same dataset as the main model to calculate pixel variance and combine scores, the system effectively detects out-of-distribution inputs, enhancing reliability and accuracy in image processing models.
Patent Information
- Application Number
- JP2022548651
- Authority / Receiving Office
- JP · JP
- Patent Type
- Patents
- Current Assignee / Owner
- Priority Date
- 2020-02-21
- Filing Date
- 2021-02-05
- Publication Date
- 2025-11-12
- Estimated Expiration
- 2041-02-05
AI Technical Summary
Existing machine-learning-enabled image processing models in critical domains like medical imaging and autonomous driving fail to provide accurate results when applied to out-of-distribution input instances, leading to misinterpretations and reduced confidence due to the generation of realistic yet anatomically incorrect images.
Implementing multiple secondary models trained on the same dataset as the main model to generate output images, calculating pixel-level variance, and combining scores to determine an overall out-of-distribution score, enabling reliable detection of input instances that are dissimilar to the training data.
This approach allows for efficient and accurate out-of-distribution detection, reducing the risk of misinterpretation by flagging anomalous inputs, thereby ensuring reliable model outputs and improving data quality in medical imaging and autonomous driving applications.
Smart Images

Figure 0007768136000001 
Figure 0007768136000002 
Figure 0007768136000003
Abstract
Description
[Technical Field]
[0001] The present invention relates to a system and a corresponding computer-implemented method for out-of-distribution detection of input instances to a model.The present invention further relates to a system and a corresponding computer-implemented method for enabling out-of-distribution detection.The present invention also relates to a computer-readable medium. [Background technology]
[0002] The processing pipeline of data coming from medical imaging devices, such as MR scanners and CT scanners, involves various complex image processing operations. For example, raw data coming from the scanner can be converted into an image to be examined by a clinician, a task known as image reconstruction. As another example, specific objects can be recognized and highlighted in images produced by medical imaging devices, a task known as semantic segmentation. Image analysis can also be used, for example, to identify lesions. In general, such image processing models generate output images from input instances, such as input images or raw scanner data represented by so-called k-space representations (in the case of MR scanners) or sinograms (in the case of CT scanners).
[0003] Given the complexity of the various image processing operations involved, the use of machine-learning-enabled models for medical image processing is becoming increasingly attractive. Such machine-learning-enabled models can be automatically trained on training datasets to perform various tasks and hold the promise of providing better results than manually designed algorithms. For example, in "Adaptive-CS-Net: FastMRI with Adaptive Intelligence" by N.Pezzotti et al. (available at https: / / arxiv.org / abs / 1912.12259 and incorporated herein by reference), a machine-learning-enabled model for reconstructing MR images from undersampled k-space data is proposed and shown to perform better than conventional methods. Another application is the denoising of low-dose CT scans. Furthermore, in various other application areas, such as autonomous driving and visual surveillance, image processing models that provide output images, such as segmentation models, are constantly improving in quality and are increasingly being deployed in practice. Summary of the Invention [Problem to be solved by the invention]
[0004] When applying such machine-learning-enabled image processing models in real life, especially in critical application domains such as medical image analysis or autonomous driving, it is important to be able to rely on the output images provided by the image processing models. However, when machine-learning-enabled image processing models are applied to input instances that are too dissimilar to the data used during training, they are generally not guaranteed to provide accurate results. Input instances that are too dissimilar are commonly referred to as "out-of-distribution" (OOD). Furthermore, something may be wrong that is not directly observable from the model's output images. For example, in medical imaging, it has been observed that some deep learning-based image processing models can generate realistic and anatomically plausible images that differ from true anatomical structures. Because artifacts appear to be true anatomical structures, radiologists may not be able to identify them as artifacts. This can lead to misinterpretations that affect diagnoses, reduced confidence in the value / quality of the product, and / or additional burden on radiologists.
[0005] Unfortunately, various known out-of-distribution detection techniques are not applicable to image processing models that have images as outputs (as opposed to classification, for example). Therefore, it is desirable to be able to perform out-of-distribution detection of input instances to image processing models. [Means for solving the problem]
[0006] Various embodiments relate to estimating whether input instances processed by a model, e.g., a deep neural network, that generates an output image were included in the dataset used for training, and therefore whether the model can be expected to produce reliable results.
[0007] According to a first aspect of the present invention, a system for out-of-distribution (OOD) detection of an input instance to a main model is proposed. According to a further aspect of the present invention, a computer-implemented method for out-of-distribution detection of an input instance to a main model is proposed. The main model can be configured to generate an output image from the input instance. The main model can be a machine-learnable model that has been trained or is being trained on a training dataset. To perform out-of-distribution detection, multiple secondary models can be used. The secondary models can be trained on the same training dataset on which the main model is trained. Given an input instance, individual secondary models can be applied to the input instance to obtain individual secondary model output images. For each pixel, a pixel out-of-distribution score can be determined as the variation between the respective values of the pixel in each secondary model output image. The pixel out-of-distribution scores can also be combined to generate an overall out-of-distribution score indicating whether the input instance is out-of-distribution with respect to the training dataset. Based on the overall out-of-distribution score, an output signal indicating whether the input instance is out-of-distribution can be generated.
[0008] To provide out-of-distribution detection, various aspects use multiple secondary models trained on the same training dataset as the main model. The multiple secondary models can be collectively referred to as an "ensemble" of secondary models. The training dataset typically includes multiple training input instances and corresponding training output images. The main model can be trained to generate corresponding training output images given training input instances. The secondary models can be trained on the same training dataset, in the sense that they can be trained to generate the same training output images given the same training input instances, or to generate outputs from such inputs in situations where the outputs and inputs are related to the original training outputs and inputs, e.g., by downscaling, channel reduction, or another image processing operation. The secondary models can use the same model architecture as the main model (optionally, the main model can be one of the secondary models), or a variation thereof, e.g., a simplification with fewer trainable parameters. For example, each secondary model can be trained by training the same trainable model based on a respective random initialization of its parameter set.
[0009] By being trained on similar data as the main model and / or using a similar architecture as the main model, the secondary models can be expected to exhibit approximately the same behavior as the main model when applied to input instances similar to the training data. Thus, at least when input instances similar to the training data are input to the secondary models, or in other words, when the input instances are in the distribution (ID), each secondary model can generally produce similar results. Therefore, the pixel-by-pixel variance of the output images of each secondary model can generally be low. However, if the input instances are out-of-distribution (OOD), each secondary model may generally have a higher pixel-by-pixel variance of their output images because they are not trained to provide similar outputs for that input instance and are trained separately. Thus, each separate pixel out-of-distribution score of a pixel in an output image can be considered a measurement of whether the input instance is OOD, and thus an overall assessment that the input instance is out-of-distribution can be obtained by combining the pixel out-of-distribution scores of one or more pixels to generate an overall out-of-distribution score.
[0010] This enables efficient and accurate out-of-distribution detection for image generative models. A relatively limited number of secondary models is sufficient to obtain an accurate overall out-of-distribution score, e.g., at most 20, at most 10, or at most 5. The inventors have already achieved good results with five secondary models. In fact, because each pixel out-of-distribution score is itself considered a measurement of whether an input instance is out-of-distribution, even with a limited number of secondary models, sufficient data may be available to determine the OOD-ness of an input instance. Furthermore, by using secondary models with fewer trainable parameters and / or fewer inputs and / or fewer outputs, the overhead of computing out-of-distribution scores associated with applying the main model can be limited. Another advantage of the provided OOD techniques is that they are largely model-agnostic, e.g., independent of the specific model architecture of the main model and / or secondary models.
[0011] According to another aspect of the present invention, an apparatus is proposed that enables out-of-distribution (OOD) detection of inputs to a main model. According to a further aspect of the present invention, a corresponding computer-implemented method is proposed. In the system and method, multiple secondary models can be trained on the same training dataset on which the main model was trained. The secondary models can then be associated with the main model to enable out-of-distribution detection. According to a further aspect of the present invention, a computer-readable medium is proposed that includes one or more instructions for performing the computer-implemented method described herein, or transient or non-transient data representing secondary models associated with the main model to enable out-of-distribution detection described herein.
[0012] In various embodiments, the out-of-distribution detection techniques described herein can be applied to medical image processing. Various models for medical image processing are known in the art and can be combined with the presented techniques. In medical image processing, the output image of the main model can also be determined from input data from a medical imaging device, such as a CT scanner or an MR scanner. Interestingly, when using data from an MR scanner, the ability to perform out-of-distribution detection as described herein can enable the safe use of undersampled k-space data (e.g., obtained by accelerated scanning), since the output image can be flagged as out-of-distribution if the input instances are not sufficiently similar to known training examples for applying the main model. This reduces the amount of data that needs to be acquired by the scanner, thereby reducing examination time while still providing reliable results. Similarly, when using data from a CT scanner, the use of out-of-distribution detection techniques can safely apply lower doses with reliable results, resulting in reduced patient exposure.
[0013] Generally, when input data from a medical imaging device is used, the input data can be signals generated by the medical imaging device. For example, in the case of an MR scanner, the signals can be represented in k-space or in image space by applying an inverse Fourier transform to the k-space representation. Similarly, in the case of a CT scanner, the signals can be represented as sinograms or in image space by applying an inverse Radon transform to the sinogram representation. When raw signals are used in this way, the main model can be, for example, a medical image reconstruction model configured to reconstruct input images from the signals. Such reconstruction models can also be called or include denoising models. For example, CT images can be reconstructed from CT signals, and MR images can be reconstructed from MR signals. However, it is also possible to train a model to reconstruct CT images from MR signals, or MR images from CT signals. Instead of using signals as input, the medical image processor model can also operate on input images reconstructed from such signals.
[0014] The main model may be a segmentation model, such as a semantic segmentation model for use in medical image processing or other application domains. The segmentation model may be configured to indicate a particular characteristic, e.g., a portion of the input instance that represents an object or other type that may be located in an image. For example, the characteristic may be any particular type of object (e.g., cyst, tumor, cell nucleus, lymphocyte, necrotic tissue, etc.) that may be present at a particular location in the input instance, or a particular characteristic that an object present in the input instance may have (e.g., dark, noisy, sharp, etc.). For example, the main model may provide, for each image pixel, an indication of whether the pixel belongs to the characteristic. The main model may also be configured to indicate, for each pixel of the input instance, the respective amount or degree of presence of the characteristic (e.g., cell or nucleus density, perfusion in tissue, etc.) at that pixel location.
[0015] The main model may be a medical image analysis model configured to determine an output image that identifies lesions in the input image. Such a model may operate on reconstructed images or signals from a medical imaging device, as appropriate. For example, the main model may be a tumor detection model for prostate or ovarian tumors, or a tumor grading model configured to assign an outlier value to portions of tumors identified in the input image.
[0016] As one skilled in the art will appreciate, the techniques provided herein are not limited to medical image processing, but can also be used in a variety of other application domains for image processing, such as autonomous driving or video surveillance.
[0017] In various aspects, an output signal indicating whether the input instance is out of the distribution can be generated based on the overall out-of-distribution score. Such an output signal can be used in various ways, for example, by a user or in further automated processes.
[0018] Optionally, the overall out-of-distribution score can be used to determine whether to apply the main model to the input instance. In such a case, the main model can be applied to the input instance only if the input instance does not indicate that it is out-of-distribution, resulting in a main model output image, which can be output, for example, to a user in a sensory perceptible manner or digitally for further automatic processing. However, it is also possible to apply the main model to input instances that are indicated to be out-of-distribution, and in such cases, for example, the output image can be shown with a warning or error that the input instance has been determined to be out-of-distribution.
[0019] Optionally, the output signal may further indicate one or more pixels of the output image that contribute to the input instance being out of distribution, e.g., a subset of pixels with the highest pixel out-of-distribution scores. For example, the indicated pixels may be pixels whose out-of-distribution scores exceed a threshold, or may be a certain percentage or number of pixels with the highest out-of-distribution scores. Indicating the contributing pixels may provide feedback that helps understand why the input instance was determined to be out of distribution and, accordingly, how this can be improved, for example, by including additional training data in the training dataset that is similar to the input instance in the respects indicated as contributing.
[0020] In general, there can be a variety of causes for an input instance to be out of distribution, and therefore a variety of remedies can be applied upon out-of-distribution detection. We now consider some causes and remedies that can be used separately or in combination.
[0021] An input instance may be out of distribution because the input instance is of low quality. For example, if the input instance is very similar to the training examples, it may represent a "routine case" for which the main model can still reliably provide a model output. In such cases, the secondary model can generally also consistently provide similar output images. However, if the input instance is not very similar to the training dataset, the secondary model may diverge, thus reducing confidence in the main model output. Therefore, the input instance can be determined to be out of distribution. For example, the input instance may be data from a CT scanner operating at too low a dose or an MR scanner operating at too high an acceleration to reliably generate an output image of the particular subject being scanned. Therefore, in some embodiments, once an input instance is determined to be out of distribution, a new measurement (e.g., a new CT or MR scan) of the input instance at a higher quality (e.g., a higher dose or lower acceleration) can be obtained, and an overall out-of-distribution score can be determined for it again. If the new input instance is not out of distribution, it can be used further, for example, presented to a user or automatically processed. Optionally, this process can be repeated for multiple quality settings.
[0022] Another possible reason for an input instance being out of distribution may be a measurement artifact in the input instance. For example, there may be a motion artifact due to the movement of the subject being scanned in the medical imaging device, or a metal artifact. Therefore, instead of or in addition to performing a new measurement with a higher quality, a new measurement with the same quality may also be performed. If the new input instance is not out of distribution, it can be used further and the previous input instance is ignored as an artifact.
[0023] Another possible cause of an input instance being out of distribution is that the object being measured by the input instance is out of distribution. For example, the object being scanned may have a pathology that is not well represented in the dataset. This cause can be recognized by determining that multiple input instances representing the same object are out of distribution, but that input instances for other objects are not. If an object is determined to be out of distribution, for example, the object can be reported to a user, e.g., a clinician, for further analysis. Optionally, upon detecting that the object is out of distribution, a different output image for the input instance can be determined using a fallback model, and the different output image can be used instead of the output image of the main model. The fallback model can be, for example, an untrainable model. For example, in the case of an MR scan, if the object is out of distribution, a known SENSE-based reconstruction can be used as a fallback.
[0024] Another possible cause of an input instance being out of distribution may be a persistent measurement problem, such as a measurement device defect or incorrect environmental conditions for the measurement. This factor can be recognized by determining that input instances representing multiple subjects are out of distribution. In such cases, the persistent measurement problem can be reported for correction.
[0025] It will be appreciated by those skilled in the art that two or more of the above-described embodiments, implementations, and / or any aspects may be combined in any manner deemed beneficial.
[0026] Modifications and variations of any computer-implemented method and / or any computer-readable medium corresponding to the described modifications and variations of the corresponding system (and vice versa) can be performed by one skilled in the art based on this specification.
[0027] These and other aspects of the invention will become apparent from and will be further elucidated with reference to the following description and the embodiments thereof described by way of example with reference to the accompanying drawings, in which: [Brief explanation of the drawings]
[0028] [Figure 1] Diagram showing a system that allows out-of-distribution (OOD) detection of inputs to the main model. [Figure 2] Diagram showing a system for out-of-distribution (OOD) detection of input instances to the main model. [Figure 3] FIG. 1 illustrates a model used in conjunction with the techniques described herein, showing a detailed example of a U-Net type model in this embodiment. [Figure 4] FIG. 10 shows a complete detailed example of determining an overall out-of-distribution score for an input instance. [Figure 5] 1 shows a computer-implemented method for out-of-distribution (OOD) detection of input instances to the main model. [Figure 6] FIG. 1 illustrates a computer-implemented method that enables out-of-distribution (OOD) detection of inputs to the main model. [Figure 7] 1 illustrates a computer-readable medium containing data. DETAILED DESCRIPTION OF THE INVENTION
[0029] It should be noted that the drawings are only schematic and are not drawn to scale. In the drawings, elements corresponding to elements already described may have the same reference numerals.
[0030] FIG. 1 illustrates an apparatus 100 for enabling out-of-distribution (OOD) detection of inputs to a main model. The main model can be configured to generate output images from input instances. The main model can be trained on a training dataset. The system 100 can have a data interface 120 and a processor subsystem 140 that can communicate internally via data communication 121. The data interface 120 can be for accessing data 030 representing a training dataset on which the main model is trained. The data interface 120 can also be for accessing the main model and / or multiple secondary models 040, as discussed in more detail below. The secondary models 040 can be used for out-of-distribution detection according to the methods described herein, for example, by the system 200 of FIG. 2. The implementation of out-of-distribution detection and the out-of-distribution detection itself can be combined into a single system or method; for example, the systems 100 and 200 can be combined into a single system. The system, e.g., its processor subsystem, can be further configured to train the main model on the training dataset 030.
[0031] The processor subsystem 140 can be configured to access data 030, 040 during operation of the system 100 using the data interface 120. For example, as shown in FIG. 1, the data interface 120 can provide access 122 to external data storage 021, which can contain the data 030, 040. Alternatively, the data 030, 040 can be accessed from internal data storage that is part of the system 100. Alternatively, the data 030, 041 can be received over a network from another entity. In general, the data interface 120 can take a variety of forms, such as a network interface to a local or wide area network, e.g., the Internet, a storage interface to internal or external data storage, etc. The data storage 021 can take any known appropriate form.
[0032] During operation of the system 100, the processor subsystem 140 can be configured to train multiple secondary models 040 using the data interface 120. The secondary models 040 can be trained on the training data set 030 on which the main model is trained. The secondary models 040 can calculate secondary model output images for inputs used in out-of-distribution detection.
[0033] The processor subsystem 140 can be further configured to associate multiple secondary models 040 with the main model to enable out-of-distribution detection. The secondary models can be associated with the main model, or more specifically with the data representing the main model, in any suitable manner, such as by including the secondary models in the model data itself, as a file header, XML element, etc., or by providing the secondary models as separate files, or in any other manner. For example, the secondary models 040 can be stored in the same data container as the main model, e.g., the same file, but can also be provided as separate secondary models related to the main model. For example, in some embodiments, the main model can link to the secondary models, e.g., by including a URL where the secondary models 040 can be accessed, or the secondary models 040 can be linked to the main model. Various other association means are equally contemplated and within the purview of those skilled in the art.
[0034] As an optional component, system 100 may include an input interface (not shown) for acquiring sensor data from a sensor, e.g., signals generated by a medical imaging device such as a CT scanner or an MR scanner. One or more training input instances of a training dataset may be based on such sensor data. Acquiring input instances via an input interface is discussed in more detail with respect to FIG. 2 , and the options described therein may apply to system 100 as well. A sensor configured to measure sensor data, such as a CT scanner or an MR scanner, may itself be part of system 100.
[0035] Various details and aspects of the operation of system 100, including any aspects thereof, are further described with reference to FIGS.
[0036] In general, system 100 can be embodied as or within a single device or apparatus, such as a laptop or desktop-based workstation or server. The device or apparatus can have one or more microprocessors executing appropriate software. For example, the processor subsystem can be embodied by a single central processing unit (CPU), but can also be embodied by a combination or system of such CPUs and / or other types of processing units. Software may be downloaded and / or stored in corresponding memory, e.g., volatile memory such as RAM, or non-volatile memory such as flash. Alternatively, functional units of the system, such as the data interface and processor subsystem, can be implemented in the device or apparatus in the form of programmable logic, e.g., as a field-programmable gate array and / or a GPU (graphics processing unit). In general, each functional unit of the system can be implemented in the form of a circuit. It should be noted that system 100 can also be implemented in a distributed manner, including multiple different devices or apparatuses, such as, for example, distributed servers in the form of cloud computing.
[0037] FIG. 2 illustrates a system 200 for out-of-distribution (OOD) detection of input instances to a main model. The main model can be configured to generate output images from the input instances. The main model can be trained on a training dataset. The system 200 can have a data interface 220 and a processor subsystem 240 that can communicate internally via data communication 221. The data interface 220 can be for accessing data 040 representing multiple secondary models for use in out-of-distribution detection. The secondary models can be trained on the same training dataset on which the main model is trained. The data interface 220 can also be for accessing the main model. The secondary model data can be obtained from a system that enables out-of-distribution detection, such as the system 100 of FIG. 1, or can be obtained by executing a method for enabling out-of-distribution detection described herein.
[0038] The processor subsystem 240 can be configured to access data 040 during operation of the system 200 using the data interface 220. For example, as shown in FIG. 2, the data interface 220 can provide access 222 to external data storage 022, which can contain the data 040. Alternatively, the data 040 can be accessed from internal data storage that is part of the system 200. Alternatively, the data 040 can be received from another entity over a network. In general, the data interface 220 can take a variety of forms, such as a network interface to a local or wide area network, e.g., the Internet, a storage interface to internal or external data storage, etc. The data storage 022 can take any known appropriate form.
[0039] During operation of the system 200, the processor subsystem 240 can be configured to obtain an input instance using the data interface 220. The processor subsystem 240 can be further configured to apply each of the plurality of quadratic models to obtain a respective quadratic model output image. The processor subsystem 240 can be further configured to determine a pixel out-of-distribution score for a pixel of each quadratic model output image. The pixel out-of-distribution score for a pixel can be calculated as the variability of each value of the pixel in each quadratic model output image. The processor subsystem 240 can further combine the calculated pixel out-of-distribution scores to generate an overall out-of-distribution score. The overall out-of-distribution score can indicate whether the input instance is out-of-distribution with respect to the training data set.
[0040] The processor subsystem 240 can further generate an output signal 225 based on the overall out-of-distribution score, where the output signal 225 indicates whether the input instance is out-of-distribution. For example, the output signal can be the overall out-of-distribution score itself or a warning that is generated if the input instance is out-of-distribution. The output signal can further indicate that one or more pixels of the output image that contribute to the input instance are out-of-distribution and provide an explanation for why the input instance is out-of-distribution, where such an output signal can be output to a rendering device or another software component for further automated processing.
[0041] Optionally, at least if the out-of-distribution score does not indicate that the input instance is out-of-distribution, the processor subsystem 240 can apply the main model to the input instance to obtain a main model output image and output the main model output image, e.g., to a user via an output interface described below or to another software component for further automatic processing. For example, to determine whether the calculated out-of-distribution score indicates OOD-likeness, the out-of-distribution score can be compared to a threshold out-of-distribution (OOD) score, e.g., a fixed score or a score associated with a quadratic model. Even if the input instances are distributed, it is not necessary to apply the main model to the input instances; for example, in some embodiments, it is possible to calculate the similarity between a dataset and a training dataset by determining out-of-distribution scores for multiple input instances of the dataset without necessarily applying the main model to these input instances.
[0042] As an optional component, the system 200 may have an input interface 260 for acquiring sensor data 223 from a sensor, e.g., from signals 224 generated by a medical imaging device such as a CT scanner or an MR scanner. An MR scanner 072 is shown in Figure 2. For example, the signals may be raw signals from the medical imaging device, e.g., represented in k-space (in the case of an MR scanner) or as a sinogram (in the case of a CT scanner), i.e., transformed into image space, e.g., by an inverse Fourier transform or an inverse Radon transform.
[0043] For example, an MR scanner can be configured to perform accelerated scans with an acceleration factor of, e.g., at least 2, at least 4, or at least 6. In such cases, while it is possible to successfully reconstruct or analyze MR data using an appropriate trained model, there is a high risk of erroneous results for inputs that are not well represented in the training data set, making out-of-distribution detection particularly important. Similar considerations apply for CT scanners operating at low doses, e.g., at most half the total dose or at most one-quarter the total dose.
[0044] However, various other types of sensors and corresponding signals 224 are possible; for example, the sensor can be a camera that generates an image, a video camera that generates video, etc. Generally, the input interface 260 can be configured for various types of sensor signals, e.g., video signals, radar / LiDAR signals, ultrasound signals, etc. The input instance on which the out-of-distribution detection is performed can be based on the sensor data 223; for example, the input instance can be equal to the sensor data 223, or any pre-processing steps can be performed on the sensor data. The sensor that measures the signal 224, such as a CT scanner or MR scanner 072, can itself also be part of the system 200.
[0045] As an optional component, the system 200 may include a display output interface 280, or any other type of output interface, for outputting the output signal 225 to a rendering device such as a display 290. For example, the display output interface 280 may generate display data 282 for the display 290, thereby causing the display 290 to render the output signal in a sensory-perceptible manner, for example, as an on-screen visualization 292. For example, if an input instance is out of distribution, an error or warning message may also be shown, for example, along with the model output of the main model. If an input instance is out of distribution, certain pixels of the secondary model output image that contribute to the out-of-distribution input instance may be rendered, for example, highlighted on the output image of the main model.
[0046] Various details and aspects of the operation of system 200, including any aspects thereof, are further described with reference to FIGS.
[0047] In general, system 200 can be embodied as or within a single device or apparatus, such as a laptop or desktop-based workstation, or a server. The device or apparatus can have one or more microprocessors executing appropriate software. For example, the processor subsystem can be embodied by a single central processing unit (CPU), but can also be embodied by a combination or system of such CPUs and / or other types of processing units. Software may be downloaded and / or stored in corresponding memory, e.g., volatile memory such as RAM, or non-volatile memory such as flash. Alternatively, functional units of the system, such as the data interface and processor subsystem, can be implemented in the device or apparatus in the form of programmable logic, e.g., a field-programmable gate array and / or a GPU (graphics processing unit). In general, each functional unit of the system can be implemented in the form of a circuit. It should be noted that system 200 can also be implemented in a distributed manner, including different devices or apparatuses, such as distributed servers in the form of cloud computing.
[0048] 3 shows a non-limiting detailed example of a model for use in the techniques described herein. For example, a main model and / or one or more secondary models for use in the techniques described herein can be constructed according to the model architecture illustrated in this figure. The particular example shown in this figure is a fully convolutional neural network, more specifically a U-Net-type model. Such models can be used, for example, for various image-to-image transformations, including the reconstruction of images from medical imaging devices.
[0049] The model shown here calculates an output image OIM 350 for an input instance IIN 330. In this example, the input instance IIN is an image represented as, for example, a matrix of size M×N×c, where the number of channels c can be c=1 for a grayscale image or c=3 for an RGB image. The number of channels does not necessarily correspond to different colors, but in this sense the input instance IIN is sometimes more generally referred to as an input volume. The output image OIM in this example is an image of the same spatial dimensions as the input instance IIN, for example, having the same width and height. The output image OIM can have the same number of channels as the input image, but the number of channels may be different.
[0050] The model herein is an example of a convolutional network, also known as a convolutional neural network. Generally, the term convolutional network can be used to refer to any neural network having at least one convolutional layer. As known in the art, a convolutional layer is a layer that operates by performing a convolution or sliding dot product operation. Thus, a convolutional layer can convert an input volume of size m×n×c into an output volume of size m′×n′×c′ using c′ filters that convolve each input volume. For example, the number of filters in a layer can be at most or at least 8, at most or at least 32, or at most or at least 128. The spatial dimensions m′×n′ of the output of a layer can differ from the spatial dimensions m×n of its input, but generally maintain spatial correspondence with the input IIN. For example, the spatial dimensions of the output of a layer can be smaller than the spatial dimensions of its input; for example, a convolutional layer can perform downsampling. The spatial dimensions of the output of a layer may be greater than the spatial dimensions of the input; for example, the layer may be a so-called "upconvolution" layer, realized by upsampling the input feature map and then applying convolution. In addition to convolutional layers, a convolutional neural network may have various other types of layers, such as one or more ReLU layers and / or one or more pooling layers. The number of convolutional layers in a convolutional network may be, for example, at least 5, or at least 10.
[0051] More specifically, the model shown in this figure is a so-called fully convolutional network: such a model transforms an input instance IIN into an output image OIM in a succession of convolutional, pooling, ReLU, and other layers, each of which maintains a spatial correspondence with the input instance.
[0052] The model in this figure is an encoder-decoder model. Generally, such a model has a contracting part (CP) 310 (also known as the "encoder path") and an expansive part (EP) 320 (also known as the "decoder path"). The contracting part (CP) can have one or more layers that generate subsequent activation volumes for the input instance I IN . The figure shows activation volumes AV1 341, AV2 342 through AVk-1 343 and AVk 344. The activation volumes can be determined from the previous activation volume by one or more layers of the model, as indicated by the arrows from 361, 362 to 363 and 364, typically max pooling followed by one or more convolutional layers with associated ReLU operations. As shown in the figure, the spatial dimensions of the activation volumes generally decrease in size throughout the contraction portion CP. For example, activation volume AV2 may have smaller spatial dimensions than activation volume AV1 (although the number of channels may actually increase), as well as for the other activation volumes shown. The activation volume AVk resulting from the contraction portion CP of the U-net model can then be processed in the expansion portion EP. The expansion portion can also have one or more layers that generate subsequent activation volumes, e.g., activation volumes AVk+1, 345 to AV2k-2, 346, and ultimately output image OIM. As noted above, activation volumes can be determined from previous activation volumes by one or more layers of the model, as indicated by arrows 365 to 366 and 367. In this case, upconvolution (e.g., upsampling followed by convolution) can typically be used, followed by one or more convolution layers with associated ReLU processing.As shown in the figure, in the extension portion EP, the spatial dimensions of the activation volumes are generally increased in size; for example, activation volume AVk-1 may have larger spatial dimensions than activation volume AVk (although the number of channels may be reduced), and similar increases may be made for the other activation volumes shown.
[0053] Interestingly, the expansion part EP can optionally have so-called skip connections, which allow the activation volumes AVi of the contraction part CP to be concatenated with the activation volumes of the expansion part EP. An encoder-decoder model with one or more skip connections is sometimes called a U-Net-type model. For example, the result of an upconvolution can be concatenated with the corresponding activation volume of the contraction part CP, and then one or more convolutions can be applied. The feature map of the contraction part can be cropped to take boundary pixels into account. For example, as shown in the figure by arrow 365′, activation volume AVk−1 can be concatenated with the upconvolution of activation volume AVk, and from the result, activation volume AVk+1 can be determined. Similarly, determination of activation volume AV2k−2 can use skip connection 366′ to activation volume AV2, determination of activation volume OIM can use skip connection 367′ to activation volume AV1, and so on.
[0054] An example of a U-net model is presented in "U-Net: Convolutional Networks for Biomedical Image Segmentation" by Ronneberger et al., available at https: / / arxiv.org / abs / 1505.04597 and incorporated herein by reference. Another known example is the so-called ResNet, or Residual Network. A detailed example of a model for MR data reconstruction that uses a U-net-type architecture in an iterative procedure and can be applied herein is presented in "Adaptive-CS-Net: FastMRI with Adaptive Intelligence" by N.Pezzotti et al.
[0055] FIG. 4 shows a detailed, non-limiting example of how to determine an overall out-of-distribution (OOD) score for an input instance to the main model, thereby performing out-of-distribution detection of the input instance.
[0056] In the figure, out-of-distribution detection can be performed in the form of calculating an overall out-of-distribution score OODS 485 (485). In one embodiment, the input I I can be fed to several secondary models S Mi 441-442. The outputs SO i 451-452 can then be used to create a so-called out-of-distribution image OODI 475. The pixel values of the out-of-distribution image OODI can also represent the variability of the corresponding pixels in the output image SO i of the model S Mi. The out-of-distribution value OODS can be obtained by reducing the out-of-distribution image OODI, for example by averaging the out-of-distribution image or by obtaining the maximum or minimum value. Once obtained, the out-of-distribution score OODS can be used to evaluate whether the input I I to the model MM is within or outside the training distribution. Such an evaluation can be a good indicator of the resulting quality of the models in the ensemble and / or the main model MM. The inputs can also be processed by a main model MM 440, which can generate a main output image OI 450 (e.g., a reconstructed MR image, a denoised CT image, etc.). As explained in more detail below, in some embodiments, the main model MM can be larger than the secondary models SMi, e.g., in terms of the number of parameters, thus enabling a higher quality main output OI while keeping the out-of-distribution score calculation OODS under control.
[0057] Specifically, an input instance IIN 430 is shown on which out-of-distribution detection is performed. As also described with respect to FIG. 3, the input instance IIN can be, for example, an input image having one, three, or another number of channels. The image can be captured by a camera, but other types of sensor data represented as an image can also be used, such as audio data or time series data of multiple sensor measurements. The input instance IIN need not be an image. For example, the input instance IIN can be a hidden representation of a feature vector from which a main model generates an output image, e.g., the generator part of a generative adversarial network (GAN), or the decoder part of an autoencoder, e.g., a variational autoencoder (VAE). In the latter part, the input instance IIN can represent synthetic data from which a synthetic output image is generated, but the input instance IIN can also represent a real-world input, e.g., by being determined as or based on the output of the encoder part of an autoencoder.
[0058] As another example, the input instance I can represent signals generated by a medical imaging device, for example, for use in medical image reconstruction or medical image analysis tasks. For example, in the case of an MR scanner, the input instance can represent scanner signals in so-called k-space or image space resulting from applying an inverse Fourier transform to k-space data. Similarly, in the case of a CT scanner, the input instance can represent scanner signals in image space as a sinogram or by applying an inverse Radon transform.
[0059] In either case, in various embodiments, the input instance IIN can include additional information in addition to the sensor / image data, e.g., metadata such as log data or patient-specific information that can help determine a more accurate output.
[0060] The figure also shows a main model MM 440. The main model can be an image processing model. In general, we refer to any model that outputs an image, e.g., a volume with width, depth, and number of channels (one, three, or any other number), but does not necessarily take an image as input, as an “image processing model.” The output image of the main model MM is shown in the figure as OI 450. The output image OIM can have a single channel, as in various segmentation models, MR or CT reconstruction models, etc., or three channels, as in various generative models, image-to-image transformation models, etc., or any other number of channels appropriate to the application of interest. The output image can be discrete, as in the case of a mask, for example, or binary, or continuous, as in the case of a generative model, for example. When the input instance is an image, the output image often has the same spatial dimensions as the input, possibly including boundary effects.
[0061] The main model MM is trained on a training dataset (not shown), typically a labeled dataset consisting of a plurality of training input instances and corresponding desired training output images (e.g., at least 1,000 to at least 1,000,000 training instances). The main model is therefore a trainable model (also known as a machine-learnable model or machine learning model). Such trainable models are generally trained by learning values for a set of trainable parameters. For example, the number of parameters of the main model may be at least 1,000, at least 100,000, or at least 10 million. Using a model that allows gradient-based optimization, e.g., a model whose parameters are continuous or differentiable, is beneficial in terms of training efficiency.
[0062] Various models can be used as the main model MM. For example, the main model can be a U-net model, or more generally, any type of encoder-decoder model, a fully convolutional neural network, a convolutional neural network, or other types of neural networks, as described with reference to FIG. 3. Such neural networks can also be used as part of a larger model, for example, in one or more iterations of an iterative model, such as in "Adaptive-CS-Net: FastMRI with Adaptive Intelligence" by N. Pezzotti et al. As mentioned above, the main model can be a generative model, for example, the generative part of a GAN or the decoder part of a VAE. In general, neural networks are also known as artificial neural networks. When using a neural network, the set of parameters can include the weights of the neural network's nodes. For example, the number of layers in the model can be at least 5 or at least 10, and the number of nodes and / or weights can be at least 1,000 or at least 10,000. Depending on the specific application, various known architectures for neural networks and other types of machine-learnable models can be used.
[0063] To perform out-of-distribution detection of input instances IIN to the main model MM, multiple secondary models SMi can be used. Secondary models SM1, 441 to SMm, 442 are shown in the figure. For example, at most or at least 5, or at most or at least 10 secondary models can be used. By applying each secondary model SMi, a respective secondary model output image SMi can be obtained, as exemplified by secondary model output images SO1, 451 to SOm, 452 in the figure.
[0064] In general, the secondary models SMi may be models trained on the same training dataset as the main model MM. Thus, when applied to input instances IIN from the training dataset, or input instances IIN similar to instances from the training dataset, the secondary models SMi can be expected to behave similarly to the main model and similar to each other. Therefore, relatively low pixel-to-pixel variability between the outputs of the secondary models can be expected. However, with input instances IIN from the training dataset, there is no such guarantee, and therefore higher pixel-to-pixel variability can be expected.
[0065] Note that achieving the above behavior does not require the secondary models SMi to use exactly the same model architecture as the main model MM (or each other), nor that the secondary models be trained on exactly the same training inputs and outputs. This observation can be exploited to improve efficiency by using secondary models SMi that are more efficient to train and / or use. For example, one, some, or all of the secondary models can be trained on downscaled training input instances. In use, such secondary models can be applied to input instances IIN by first downscaling the input instances. One, some, or all secondary models can alternatively or additionally be trained on downscaled training output images, thus generating smaller output images that can be later upscaled, as needed, during use. For example, the input and / or output can be downscaled by at least a factor of 2, or at least a factor of 4.
[0066] In general, the use of smaller inputs and / or outputs allows for the use of smaller secondary models, e.g., which may have fewer trainable parameters than the main model, leading to reduced storage and computational requirements. Alternatively, the secondary model can be simplified to reduce the number of trainable parameters, e.g., by using a neural network with fewer layers, or by using an iterative model with fewer iterations. For example, the secondary model may have up to half, up to 25%, or up to 10% of the number of trainable parameters of the main model. For example, if the main model is an iterative model, the secondary model may also be an iterative model with up to half, up to 25%, or up to 10% of the number of iterations of the main model, instead of, or in addition to, reducing some parameters used in a single iteration. For example, the inventors obtained good results using a secondary model with three iterations compared to a main model with 15 iterations.
[0067] In some embodiments, the secondary models SMi have a common model architecture, e.g., they can be trained by initializing a set of parameters of the common model and optimizing the set of parameters based on the initialization, but using different random initializations. This common architecture may be the same as or different from the architecture of the main model MM. This technique has the advantage of being generally applicable and suitable for parallelization, both in the training and in the use of the secondary models.
[0068] Although the main model MM is shown in this figure separately from the secondary models SMi, the main model can, for example, be used as one of the secondary models both to determine the main model output and to determine the secondary model output. The main model MM can also be composed of some or all of the secondary models, and for example the main model output can be determined based on the outputs of one or more of the secondary models SMi, for example by averaging or another type of combination operation.
[0069] In processing PXS 470, the secondary model output images SOi can be used to calculate pixel outlier scores for the pixels of each secondary model output image SOi. For example, a pixel outlier score for the top left pixel having coordinates of each image can be calculated, and similarly for other pixels. In some embodiments, a pixel outlier score can be calculated for each pixel. However, it is not necessary to calculate a pixel outlier score for every pixel; alternatively, for example, for reasons of efficiency, a sample of pixels can be obtained. The pixel outlier scores for some or all of the pixels can be considered to form an image that can itself be referred to as an "outlier image" for the input instance IIN. The outlier image OODI 475 is shown in the figure.
[0070] Optionally, the processing PXS may include resizing (e.g., downscaling the images to a minimum size) and / or channel reduction to make the secondary model output images SOi correspond to their size (e.g., converting the images to grayscale).
[0071] The pixel out-of-range score for a pixel can also be calculated as the variability of each value of the pixel in each secondary model output image. For example, the variability can be calculated as a measure of the variability of a vector including pixel values for the secondary output image SOi, e.g., grayscale pixel values between 0 and 1, discretized pixel values, etc. Any suitable measure of variability can be employed, e.g., variance or standard deviation, Shannon entropy, etc. In the case of a multi-channel secondary output image, the output image can be converted to a single-channel image, e.g., converted to grayscale. It is also possible to calculate a pixel out-of-range score for each channel, in which case the pixel out-of-range score can be obtained by combining the scores for each channel (e.g., as the highest, lowest, or average) or by considering the vector of scores for each channel as the pixel out-of-range score.
[0072] After calculating the pixel out-of-distribution scores OODI for the pixels of the secondary model output image SOi, these scores can be combined in a combination process CMB 480 to generate an overall out-of-distribution score OODS 485. The overall out-of-distribution score OODS can indicate whether the input instance is out-of-distribution with respect to the training dataset.
[0073] Various methods of using the overall out-of-distribution score OODS are possible. For example, the maximum, minimum, or average value can be used, or any other calculation that generally provides a higher score for a higher pixel out-of-distribution score OODI, such as a non-decreasing or increasing function of each pixel out-of-distribution score, can also be used. Different score types provide different indicators of OOD-ness that may be useful in different situations. The maximum value can be used as a high-assurance option to ensure that there are no parts of the output image from which the secondary models diverge and therefore ensure that each part of the main model's output image is reliable. The average and minimum values can be used to obtain a more global measure of the reliability of the main model output. Because different measures provide different types of information about the OOD-ness of the input instance IIN, multiple out-of-distribution score values can also be output; for example, the overall out-of-distribution score can include each of the component out-of-distribution score values, such as the maximum and average.
[0074] Based on the global out-of-distribution score OODS, a threshold out-of-distribution score can be applied to determine whether an input instance is out-of-distribution. In some embodiments, as part of training the secondary model, the threshold score can be automatically calculated based on the out-of-distribution scores calculated for multiple input instances for the main model. For example, the threshold score can be determined as a threshold for a statistical test of the global out-of-distribution score belonging to the statistical distribution of out-of-distribution scores of the in-distribution input instances. For example, the threshold score can be calculated as an optimal threshold score for the training set (including in-distribution samples) and the holdout set (including out-of-distribution samples).
[0075] The secondary model can be trained using various known techniques, as known per se in the art. Optionally, the main model is trained by the same system as the secondary model, but a pre-trained main model can also be taken and the secondary model trained to enable out-of-distribution detection for it. Typically, training is performed using a probabilistic technique such as stochastic gradient descent, for example using the Adam optimizer as disclosed in Kingma and Ba, "Adam: A Method for Stochastic Optimization" (available at https: / / arxiv.org / abs / 1412.6980, incorporated herein by reference). As is known, such optimization methods can be heuristic and / or reach a local optimum. Training can be performed instance-by-instance or in batches, for example on at most or at least 64, or at most or at least 256 instances.
[0076] 5 shows a block diagram of a computer-implemented method 500 for out-of-distribution (OOD) detection of input instances to a main model. The main model can be configured to generate output images from the input instances. The main model can also be trained on a training dataset. Method 500 may correspond to the processing of system 200 of FIG. 2. However, this is not a limitation, and method 500 can also be performed using another system, apparatus, or device.
[0077] The method 500 may include accessing 510 data representing multiple secondary models for use in out-of-distribution detection in an operation titled "Accessing Secondary Models." The secondary models may be trained on the same training data set on which the main model is trained.
[0078] The method 500 may include obtaining 520 an input instance in an operation entitled "Obtaining an Input Instance."
[0079] Method 500 may include applying 530 each of the plurality of quadratic models to obtain a respective quadratic model output image in an operation entitled "Applying Quadratic Models."
[0080] Method 500 may include calculating 540 a pixel out-of-distribution score for a pixel of each secondary model output image in an operation entitled "Determining a Pixel Out-of-Distribution Score." The pixel out-of-distribution score for a pixel may also be determined as the variability of each value of the pixel in each secondary model output image.
[0081] Method 500 may include combining 550 the calculated pixel out-of-distribution scores to generate an overall out-of-distribution score in an operation entitled "Combining to Generate an Overall Out-of-Distribution Score." The overall out-of-distribution score may indicate whether the input instance is out-of-distribution with respect to the training dataset.
[0082] The method 500 may include, in an operation entitled "Generating an Output Signal," generating 560 an output signal based on the overall out-of-distribution score, the output signal indicating whether the input instance is out-of-distribution.
[0083] 6 shows a block diagram of a computer-implemented method 600 that enables out-of-distribution (OOD) detection of inputs to a main model. The main model can be configured to generate output images from input instances. The main model can also be trained on a training dataset. Method 600 may correspond to the operations of system 100 of FIG. 1. However, this is not a limitation, and method 600 can also be performed using another system, apparatus, or device.
[0084] Method 600 may include, in an operation entitled "Accessing a Training Dataset," accessing 610 data representing the training dataset on which the main model was trained.
[0085] Method 600 may include training 620 multiple secondary models in an operation entitled "Training Secondary Models." The secondary models may be trained on the training dataset on which the main model is trained. The secondary models may, for example, be for determining secondary model output images of inputs for use in out-of-distribution detection according to method 500.
[0086] Method 600 may include associating 630 multiple secondary models with the main model to enable out-of-distribution detection in an operation titled "Associating Models with the Main Model."
[0087] It will be appreciated that the operations of method 500 of Figure 5 and method 600 of Figure 6 can be performed in any suitable order, e.g., sequentially, simultaneously, or a combination thereof, subject to a particular order being required where applicable, e.g., by input / output relationships. Some or all of the methods can also be combined, e.g., method 500 can be used to perform out-of-distribution detection, followed sequentially by method 600 to enable out-of-distribution detection.
[0088] The method can be implemented on a computer, as a computer-implemented method, as dedicated hardware, or as a combination of both. Also, as shown in FIG. 7 , instructions for a computer, e.g., executable code, can be stored on a computer-readable medium 700, e.g., in the form of a series of machine-readable physical marks 710 and / or as a series of elements having different electrical, e.g., magnetic, or optical properties or values. The executable code can be stored transiently or non-transiently. Examples of computer-readable media include memory devices, optical storage devices, integrated circuits, servers, online software, etc. FIG. 7 illustrates an optical disk 700. Alternatively, the computer-readable medium 700 can include transient or non-transient data 710 representing multiple secondary models. The secondary models can be associated with a main model to enable out-of-distribution detection. The main model can be configured to generate output images from input instances. The main model can be trained on a training dataset. The secondary models can be trained on the training dataset on which the main model is trained. The secondary models can be configured to determine secondary model output images for inputs for use in out-of-distribution detection. The secondary model can be obtained according to the computer implementation 600. The secondary model can be configured for use in the computer-implemented method 500.
[0089] No examples, embodiments, or optional features, whether non-limiting or not, should be understood as limiting the invention as recited in the claims.
[0090] It should be noted that the above-described embodiments illustrate rather than limit the present invention, and that those skilled in the art will be able to design many alternative embodiments without departing from the scope of the appended claims. In the claims, any reference signs placed between parentheses shall not be construed as limiting the claim. Use of the verb "comprise" and its conjugations does not exclude the presence of elements or steps other than those stated in a claim. The article "a" or "an" preceding an element does not exclude the presence of a plurality of such elements. The use of phrases such as "at least one of" preceding a list or group of elements denotes selection of all or any subset of the elements from the list or group. For example, the phrase "at least one of A, B, and C" shall be understood as including A only, B only, C only, both A and B, both A and C, both B and C, or all of A, B, and C. The present invention can be implemented by means of hardware comprising several distinct elements, and by means of a suitably programmed computer. In a device claim enumerating several means, several of these means may be embodied by one and the same item of hardware. The mere fact that certain measures are recited in mutually different dependent claims does not indicate that a combination of these measures cannot be used to advantage.
Claims
1. 1. A system for out-of-distribution detection of input instances to a main model, said main model being trained on a training dataset, said main model being configured to generate output images from said input instances, said out-of-distribution representing dissimilarity between the input instances and the training dataset, said dissimilarity being represented by an overall pixel out-of-distribution score higher than a predefined threshold score, said system comprising: a data interface for accessing data representing a plurality of secondary models used for out-of-distribution detection of the main model, the secondary models being trained on the same training dataset as the main model; and a processor subsystem; wherein the processor subsystem comprises: obtaining an input instance of the main model; applying each of the plurality of secondary models to an input instance of the main model to obtain a respective secondary model output image; determining a pixel outlier score for pixels of each of the secondary model output images, the pixel outlier score being calculated as the variance between respective numerical values of pixels in each of the secondary model output images; combining the calculated pixel out-of-distribution scores to generate an overall out-of-distribution score, the overall out-of-distribution score indicating whether the input instance is out-of-distribution with respect to the training dataset; generating an output signal based on the overall out-of-distribution score, the output signal indicating whether the input instance is out-of-distribution; A system configured to run
2. The system of claim 1 , wherein the main model is configured to determine the output image from input data of a medical imaging device.
3. The system of claim 2 , wherein the main model is a medical image reconstruction model configured to reconstruct the output image from signals produced by the medical imaging device.
4. The system of claim 2 , wherein the main model is a medical image analysis model configured to determine an output image that locates a lesion in an input image.
5. The system of claim 1 , wherein a quadratic model is trained on downscaled training input instances and / or downscaled training output images of the training dataset.
6. The system of claim 1 , wherein a secondary model has fewer trained parameters than the main model.
7. 7. The system of claim 1, wherein the processor subsystem is further configured to: apply the main model to the input instance to obtain a main model output image, and output the main model output image, at least if the overall out-of-distribution score does not indicate that the input instance is out-of-distribution.
8. The system of claim 1 , wherein the output signal further indicates that one or more pixels of the secondary model output image that contribute to a previous input instance are out of distribution.
9. 9. The system of claim 1, further comprising an output interface for outputting the output signal to a rendering device that renders the output signal in a manner that is perceptible to a user.
10. 1. A system for enabling out-of-distribution detection of input instances to a main model, said main model being trained on a training dataset, said main model being configured to generate output images from said input instances, said out-of-distribution representing a dissimilarity between said input instances and said training dataset, said dissimilarity being represented by an overall pixel out-of-distribution score higher than a predefined threshold score, said system comprising: a data interface for accessing data representing a training dataset on which the main model is trained; a processor subsystem; wherein the processor subsystem comprises: training a plurality of secondary models, the secondary models being trained on the same training dataset as the main model, the secondary models being configured to determine secondary model output images for input instances used in the out-of-distribution detection; associating the plurality of secondary models with the main model to enable the out-of-distribution detection; A system configured to run
11. The system is further configured to determine a threshold out-of-distribution score for the out-of-distribution detection and associate the threshold out-of-distribution score with the main model, the threshold out-of-distribution score being determined based on out-of-distribution scores of a plurality of input instances to the main model, the out-of-distribution scores of the input instances being: applying each of the plurality of quadratic models to obtain a respective quadratic model output image; calculating a pixel outlier score for each pixel of each secondary model output image, the pixel outlier score for a pixel being calculated as a variance between respective numerical values of the pixel in each secondary model output image; combining the calculated pixel out-of-distribution scores to generate an overall out-of-distribution score, the overall out-of-distribution score indicating whether the input instance is out-of-distribution with respect to the training dataset; The system of claim 10 , wherein the determination is made by executing:
12. 12. The system of claim 10 or 11, wherein the quadratic models are trained by initializing a set of parameters of a trainable model and optimizing the set of parameters based on the initialization, and each of the quadratic models is trained by training the same trainable model based on a respective random initialization.
13. 1. A computer-implemented method for out-of-distribution detection of input instances to a main model, the main model being trained on a training dataset, the main model being configured to generate output images from the input instances, out-of-distribution representing dissimilarity between the input instances and the training dataset, the dissimilarity being represented by an overall pixel out-of-distribution score higher than a predefined threshold score, the method comprising: accessing data representing a plurality of secondary models for use in out-of-distribution detection of the main model, the secondary models being trained on the same training data set as the main model; obtaining an input instance of the main model; applying each of the plurality of secondary models to the input instance of the main model to obtain a respective secondary model output image; calculating a pixel outlier score for each said secondary model output image pixel, the pixel outlier score being calculated as the variance between respective numerical values of pixels in each said secondary model output image; combining the calculated pixel out-of-distribution scores to generate an overall out-of-distribution score, the overall out-of-distribution score indicating whether the input instance is out-of-distribution with respect to the training dataset; generating an output signal based on the overall out-of-distribution score, the output signal indicating whether the input instance is out-of-distribution; A method having the following.
14. 1. A computer-implemented method for enabling out-of-distribution detection of inputs to a main model, said main model being trained on a training dataset, said main model being configured to generate output images from input instances, said out-of-distribution representing dissimilarity between said input instances and said training dataset, said method comprising: accessing data representing a training dataset on which the main model is trained; training a plurality of secondary models, the secondary models being trained on the training dataset on which the main model is trained, the secondary models being configured to determine secondary model output images for input instances used for out-of-distribution detection; associating the plurality of secondary models with the main model to enable the out-of-distribution detection; A method having the following.
15. instructions that, when executed by a processor system, cause the processor system to perform the computer-implemented method of claim 13 or claim 14; A plurality of second-order models; wherein the secondary model is associated with the main model to enable out-of-distribution detection, the main model being trained on a training dataset, the main model being configured to generate output images from input instances, and the secondary model being trained on the training dataset on which the main model is trained, the secondary model being configured to determine secondary model output images for input instances used in out-of-distribution detection.
Citation Information
Patent Citations
Information processing device, information processing method, and program
JP2018026122A
Determination device, analysis system, determination method and determination program
JP2018190127A
Evaluation device, motion control device, evaluation method, and evaluation program
JP2019139277A
Capsules for image analysis
US20190370972A1
Machine learning method and device, program, learned model, and identification device
WO2019167884A1