Thursday, 26 June 2008

Harmonic Stars

One of my pet projects is a web based system for recording all the DVDs and Videos that we own. The original aim of the system was to simply record everything we owned so that we would know what there was to watch. As with everything else software related that I start, I have spent more time on it then is reasonable and so it has grown to be so much more than a simple database. One of the more useful features that I added was a way for a group of people to figure out what they should sit down and watch. To improve the suggestions the system was making I decided to add star ratings. Now each user can assign a star rating to each film which is then used to make suggestions for the group. The problem arose when I was trying to work out the combined star rating for a group of users.

The easiest approach is simply to average the star ratings from the members of a group. For now I'm ignoring the issue of what to do with a film which one or more users hasn't rated (in the final system they are never actually suggested so the problem nicely disappears). Unfortunately simply averaging doesn't produce nice results. Lets assume (for the sake of simplicity) that there are just two users, then if they both give a film 3 out of 5 stars then it gets an average of 3 stars. This seems totally reasonable. If, however, one gives it a 5 and the others gives it a 1, then it also recieves an average star rating of 3. Clearly this isn't right. There is no way we should be suggesting a film that one person loves and one hates.

So my requirements are a function that 1) ranges between 0 and 5 so that I can match the combined star ratings onto the same scale as the individual ratings and 2) gives a combined star rating that matches my logical intuition of what should happen. Well that shouldn't be too hard to find!

After a long period of simply trying every averaging function in Excel I came to the conclusion that the clear winner was the harmonic mean. That is the reciprical of the arithmetic mean of the recipricals or for n positive real numbers:



I have no clear idea as to why the harmonic mean works but it does. Just to show you the difference here is all possible combinations of ratings from two users, showing both the average and harmonic mean to 2 decimal places and the final joint rating (the harmonic mean to 0 decimal places)

Rating 1Rating 2AverageHarmonic MeanJoint Rating
5.005.00
4.504.44
4.504.44
4.004.00
4.003.75
4.003.75
3.503.43
3.503.43
3.003.00
3.502.86
3.502.86
3.002.67
3.002.67
2.502.40
2.502.40
2.002.00
3.001.67
3.001.67
2.501.60
2.501.60
2.001.50
2.001.50
1.501.33
1.501.33
1.001.00

Note that the list is sorted by the harmonic mean so you can see by looking at the average column that it not only gives different numerical results but also would give a different ordering.

Maybe some of the more mathematically inclined of my readers may care to enlighten us all as to why the harmonic mean seems to be a good fit.

Not long after starting to write this post I came across another place where the same problem occurs: peer reviewing papers. I'm currently organising a workshop and so had to decide which of the submitted papers to accept based upon 3 reviews per paper. The system we use to help manage the process (START V2) by default simply averages the overall scores assigned to a paper. This though, suffers from the same problem as the star ratings and should probably be changed to use the harmonic mean. Fortunately in the case of my workshop while the order of the papers did change it didn't make any difference to which papers I accepted, but something to remember for the future.
26 June 2008 at 22:36 , Rob said...

Can't offer an explanation but I agree with the intuitive assessment that the harmonic mean is the best of the options discussed. Of course the real challenge is in predicting what a user will think of a movie they haven't yet seen. I seem to remember hearing that there is public discussion of some of the algorithms competing in the netflix prize (http://www.netflixprize.com/), if you could make use of some of that then I can look forward to even better viewing when I drop around randomly for an evening!!

Cheers,

Rob

27 June 2008 at 06:25 , Scriptor Senex said...

Duh! (And that's one of the more intelligent things I thought...). I shall definietly leave harmonic means to others.

1 July 2008 at 02:31 , Mark said...

Rob, could you leave a comment with pointers to any of the Netflix papers you come across in case I haven't seen them already.

No promises though!

Post a Comment