/
atom.xml
41 lines (35 loc) · 7.04 KB
/
atom.xml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
<?xml version="1.0" encoding="utf-8"?>
<feed xmlns="http://www.w3.org/2005/Atom">
<title>
BASIC blog full text feed </title>
<link href="blog/atom.xml" rel="self" />
<link href="http://kyleteague.com/"/>
<updated>2011-10-18T10:14:37Z</updated>
<id>http://kyleteague.com/blog/atom.xml/</id>
<entry>
<title type="html">How We Finished in the Top 10 in the KDD Cup</title>
<author><name>Kyle Teague</name></author>
<link href="http://kyleteague.com/blog/top-10-kdd-cup.html"/>
<updated>2011-10-18T06:51:00Z</updated>
<published>2011-10-18T06:51:00Z</published>
<id>http://kyleteague.com/blog/top-10-kdd-cup.html</id>
<category scheme="http://kyleteague.com/blog/tags"
term="kdd"
label="Kdd" />
<category scheme="http://kyleteague.com/blog/tags"
term="square counting"
label="Square Counting" />
<content type="html">
<h2>Introduction</h2>
<p>Earlier this year my colleagues and I participated in both tracks of the <a href="http://kddcup.yahoo.com">Yahoo! <span class="caps">KDD</span> Cup</a>. To give some background, the <span class="caps">KDD</span> Cup is the biggest annual machine learning competition (at least unpaid). This year Yahoo! sponsored the event and released a dataset of around 250 million music ratings on genres, artists, albums, and tracks (&#8220;items&#8221;). There were two different tracks in the competition. Track 1 was concerned with predicting the ratings a user gave on a held-out test set (regression) and Track 2 was focused on predicting whether a user would love or hate a given item (binary&nbsp;classification).</p>
<p>I ended up leaving the company sponsoring my work early in the competition to join <a href="http://getglue.com">GetGlue</a> as their data scientist, effectively ending my contributions code-wise. I was around #3 or #4 in Track 1 at the time with a blended matrix factorization algorithm. It looked at a few temporal and hierarchical features and was solved using a simple gradient descent algorithm with an adaptive learning rate. It was still good enough for 24th place overall. I don&#8217;t know the exact number of teams, but I heard unofficially that it topped over 1,000. My colleague, <a href="http://www.ee.ucla.edu/~jskong">Joseph Kong</a> stayed on and continued to work part time on Track 2 where we (mostly he) <a href="http://kddcup.yahoo.com/leaderboard.php?track=2&amp;n=100">placed 9th</a>. Unlike other competitors ahead of us though, we did not use an ensemble stack (other than stacking our own algorithm with parameter variations). The best single method was around 3.7%. Most of the top <span class="caps">MF</span>-<span class="caps">BPR</span> algorithms could only achieve about what we did on the Test1&nbsp;set.</p>
<h2>A Primer on Square&nbsp;Counting</h2>
<p>We differed from most of the competitors due to Joseph coming up with a radical &#8220;square counting&#8221; algorithm. He was inspired by triangle counting in graph mining.
<img class="cent" src="/media/images/square_configs.png" alt="Square Configurations" />
So let&#8217;s say we want to see if you&#8217;ll like song A. You and a friend both love song B and you and another friend hate song C. The first friend loves A, but the second friend hates it. The situations with the two friends are modeled in the above figure, specifically in configurations 7 and 0. Now let&#8217;s expand this to thousands of friends. We can actually count the configurations for each person/song pair we want to predict reducing each pair into a feature vector of length 2<sup>3</sup>. The feature vectors can then be passed into your classic classification algorithm of choice. Of course, there are other subtleties like normalizing the counts based on the degrees of the nodes and other enhancements we made to get down to a 4.6% error rate, but the basic idea is pretty simple. There is also the problem of designing an efficient algorithm to actually do the square counting. Oh, and if you&#8217;re wondering, our research shows that you&#8217;ll most likely hate song A &#8212; hate is a better&nbsp;predictor.</p>
<h2>Conclusion</h2>
<p>Square counting is an effective algorithm in a bipartite graph with a binary rating system. We have ideas on how to leverage square counting for regression (think soft-max) and other graph types, but this is simply the first iteration. It is essentially a feature extraction stage which transforms the problem from a sparse ratings matrix into a traditional classification problem. The nice part is you can easily parallelize it, then feed the output into any number of machine learning frameworks, which require far less processing time than would be required for a matrix factorization method. Using the <a href="http://cran.r-project.org/web/packages/gbm">gbm package</a> in R worked just fine for our use&nbsp;case.</p>
<p><a href="http://kddcup.yahoo.com/pdf/Track2-KKTsLearningMachine-Paper.pdf">Read the paper here</a> or view the&nbsp;slides:</p>
<div style="width:425px" id="__ss_9741700"><strong style="display:block;margin:12px 0 4px"><a href="http://www.slideshare.net/kyleteague1/just-count-the-lovehate-squares" title="Just Count the Love-Hate Squares">Just Count the Love-Hate Squares</a></strong><object id="__sse9741700" width="425" height="355"><param name="movie" value="http://static.slidesharecdn.com/swf/ssplayer2.swf?doc=track2-kktslearningmachine-slides-111018014236-phpapp01&stripped_title=just-count-the-lovehate-squares&userName=kyleteague1" /><param name="allowFullScreen" value="true"/><param name="allowScriptAccess" value="always"/><embed name="__sse9741700" src="http://static.slidesharecdn.com/swf/ssplayer2.swf?doc=track2-kktslearningmachine-slides-111018014236-phpapp01&stripped_title=just-count-the-lovehate-squares&userName=kyleteague1" type="application/x-shockwave-flash" allowscriptaccess="always" allowfullscreen="true" width="425" height="355"></embed></object><div style="padding:5px 0 12px">View more <a href="http://www.slideshare.net/">presentations</a> from <a href="http://www.slideshare.net/kyleteague1">Kyle Teague</a>.</div></div> </content>
</entry>
</feed>