0
$\begingroup$

I have a bunch of images taken from a camera showing a pipe and would like to detect if the pipe is leaking or not. There are very few examples of leaking pipe in the data set. So considering this problem as a supervised learning problem, I think that it may not give us good result due to imbalanced data. I'm thinking of using autoencoder and considering it as an outlier detection problem.

I'm new to deep learning so I'd like to know how the architecture of my neural network should look like. Should I have some convolutional layers first and then an autoencoder or I should only have autoencoder? What would be the best DL library for such a use case? I also thinking of using only the photos which do not have any leak for the training phase, is that OK?

$\endgroup$
  • $\begingroup$ Why exactly do you think that supervised learning is not a approach to this problem? What are the shape of your classes in the dataset that is creating imbalance? Add that to the question and hope that will help provide exact solution. $\endgroup$ – thanatoz Dec 2 '18 at 22:56
0
$\begingroup$

If you use supervised learning you need to weighted you labels (here).

Otherwise I strongly recommend to exclude every leaking image for a training dataset for an CAE. If you have the anomaly in your trainign dataset, your CAE will learn it as normal. This is actually an active research field, and not trivial for real datsets, but if you have a static image of the pipe, it should be easy. I would suggest that you use traditional CV for it and not an DL model, because you can interpret and fine tune the result.

| improve this answer | |
$\endgroup$
0
$\begingroup$

Your explanation is not full enought. If you have same images ,with same background,it's mean that you can use simple image diff, to find changes and detect leaking.

| improve this answer | |
$\endgroup$
0
$\begingroup$

One idea is to train your autoencoder only one normal images which will allow you to represent in a lower dimension space the problem. Your image would be encoded as follows, for $x \in \mathbb{R}^{W \times H \times C}$

$$ f(x) = \tilde{x} \in \mathbb{R}^{1 \times d} $$

Then since you are in a lower dimension space you can use a distance-based rule: your normal cases should accumulate in some ball:

$$ \{ ||x-u|| < \varepsilon \} $$

where $u$ would be the center (average of all your normal training cases). Then use the train model to encode your images at prediction time. Then if a new point is inside the ball it is a normal case, if it is outside it is an outlier.

That's just an idea, I think there are plenty of possible ways to approach the problem.

| improve this answer | |
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.