I am trying to formulate and solve the following problem of image mutation. Suppose I am trying to insert an object image into a "background" image of several objects, and I will need to look for a "sweet spot" to insert the image:

enter image description here

I am tentatively trying to formulate the problem into a reinforcement learning process, with the following elements:

0. initial stage:

  • a background image where the location of objects within the image has been marked (let's suppose we have a perfect object detector)

  • another image of a new object, let's say, a human

1. action space:

  • location (x, y) for the object image to be inserted; in that sense the action space is quite large.

2. environment:

  • each step I will have a new image to "learn from".

  • An oracle function F returns 1 or 0 (roughly one computation of F takes 30 seconds). This function tells me the latest synthesized image hits the "sweet spot" or not (1 means hit). If so, I will stop the search and return the image.

3. constraint:

the newly inserted object shouldn't overlap with the original objects in the figure.

While my gut feeling is that this problem is somehow similar to the classic "maze escape" problem which can be solved well with reinforcement learning, the action space seems quite large in this problem.

So here are my questions:

  1. In case I would like to formulate this "beautify" image problem into a "deep" reinforcement learning problem, how can I learn from such large action space? Or is it really suitable for a reinforcement learning process?

  2. Can I somehow subsume the "non-overlapping" constraint into the oracle function F? If so, how should I decide the reward score? Any principled or empirical way of deciding so?


The purpose of Reinforcement Learning is to maximize some notion of cumulative reward, leading me to the point (1) : as far as I understand, there is no timesteps in your problem and the "reward" is immediate. Thus, I don't think reinforcement is suitable here.

On an other hand, in supervised learning, linear regression is the task of approximating a mapping function (f) from input variables (X) to a continuous output variable (y). It has much in common with your case. If I am not wrong, you are trying to approximate a function that maps image data to (x,y) coordinates (the "sweet spot"). So, I think regression would be a better way to go for you.

You could either generate a dataset at first, with images data and associated (x,y) coordinates for only spots validated by your function F, and then train a regression predictive model. Or you could train your model with online learning, by generating batch of images and sweet spot coordinates at each step.

Concerning the point (2), it will highly depend on how is made your F function. Since overlapping spots cannot be sweet spots, the simplest would be to make your F function return 0 for those spots.

| improve this answer | |

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.