<!DOCTYPE article PUBLIC "-//NLM//DTD JATS (Z39.96) Journal Archiving and Interchange DTD v1.0 20120330//EN" "JATS-archivearticle1.dtd">
<article xmlns:xlink="http://www.w3.org/1999/xlink">
  <front>
    <journal-meta />
    <article-meta>
      <title-group>
        <article-title>RBPR: Role-based Bayesian Personalized Ranking for Heterogeneous One-Class Collaborative Filtering</article-title>
      </title-group>
      <contrib-group>
        <contrib contrib-type="author">
          <string-name>CCS Concepts</string-name>
          <xref ref-type="aff" rid="aff0">0</xref>
          <xref ref-type="aff" rid="aff1">1</xref>
        </contrib>
        <aff id="aff0">
          <label>0</label>
          <institution>Xiaogang Peng</institution>
          ,
          <addr-line>Yaofeng Chen, Yuchao Duan, Weike Pan</addr-line>
        </aff>
        <aff id="aff1">
          <label>1</label>
          <institution>Information systems → Personalization; •Humancentered computing → Collaborative filtering</institution>
        </aff>
      </contrib-group>
      <abstract>
        <p>Heterogeneous one-class collaborative filtering (HOCCF) is a recently studied important recommendation problem, which consists of different types of users' one-class feedback such as browses and purchases. In HOCCF, we aim to fully exploit the heterogenous feedback and learn users' preferences so as to make a personalized and ranking-oriented recommendation for each user. For HOCCF, we can apply existing solutions for OCCF with purchases only such as Bayesian personalized ranking (BPR) or make use of both browses and purchases such as transfer via joint similarity learning (TJSL). However, BPR may be not very accurate due to the ignorance of browses, and TJSL may be not very efficient due to the mechanism of joint similarity learning and base model aggregation. In this paper, we propose a novel perspective for the different types of one-class feedback via users' different roles, i.e., browser and purchaser. Specifically, we design a two-stage role-based preference learning framework, i.e., role-based Bayesian personalized ranking (RBPR). In RBPR, we first digest the combined one-class feedback as a browser to find the candidate items that a user will browse, and then we exploit the purchase feedback to refine the candidate list as a purchaser. Empirical results on five public datasets show that our RBPR is an efficient and accurate recommendation algorithm for HOCCF as compared with the state-of-the-art methods such as BPR and TJSL.</p>
      </abstract>
    </article-meta>
  </front>
  <body>
    <sec id="sec-1">
      <title>INTRODUCTION</title>
      <p>
        Intelligent recommendation systems and technology have
played a more and more important role in various real-world
applications, with a wide spectrum of entertainment, social
and professional services. Some recent work show that one
important line of research have gradually transferred from
collaborative filtering (CF) with numerical ratings to
oneclass CF (OCCF) with homogeneous one-class feedback such
as purchases [
        <xref ref-type="bibr" rid="ref2">2</xref>
        ] and heterogeneous OCCF (HOCCF) with
more than one types of one-class feedback such as browses
and purchases [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]. In this paper, we focus on the problem
setting of HOCCF, which is very common in real industry
scenarios.
      </p>
      <p>
        The main challenge of HOCCF is the heterogeneity of the
two different types of one-class feedback, since a user’s
preference behind a purchase action may be different from that
of a browse action. In a very recent work [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ], a
similarity learning algorithm is proposed for this challenge, which
aims to combine browses and purchases in a principled way.
The improved performance in [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ] shows the
complementarity of browses to the well exploited feedback of purchases
in OCCF models [
        <xref ref-type="bibr" rid="ref1 ref4">1, 4</xref>
        ]. However, the proposed algorithm,
i.e., transfer via joint similarity learning (TJSL) [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ], may be
not efficient enough for large datasets due to the complex
prediction rule and base model ensemble.
      </p>
      <p>
        In this paper, we interpret the HOCCF problem from a
novel view of users’ roles, i.e., a purchaser (as reflected in
a purchase feedback) is converted from a browser in a
sequential manner. Based on this perspective, we propose
a two-stage framework, including browser-based preference
learning and purchaser-based preference learning. Those
two preference learning tasks are connected via a candidate
list of items that a user will browse, which is assumed to
contain the potential items that a user will finally purchase.
In each of the two tasks, we apply the seminal work for
homogeneous one-class feedback, i.e., Bayesian personalized
ranking (BPR) [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ], and for this reason, we call our approach
role-based BPR (RBPR).
      </p>
      <p>In our empirical studies, we compare our RBPR with the
state-of-the-art methods of BPR and TJSL using various
ranking-oriented evaluation metrics on five public datasets.
The studies show that our RBPR is able to produce
competitive recommendations efficiently. We list our main
contributions as follows: (i) we propose a novel and generic
staged role-based preference learning framework, which is a
frustratingly easy, scalable and effective solution for
collaborative ranking with heterogeneous one-class feedback; and
(ii) we conduct extensive empirical studies and obtain very
promising results.</p>
    </sec>
    <sec id="sec-2">
      <title>ROLE-BASED BAYESIAN PERSONALIZED</title>
    </sec>
    <sec id="sec-3">
      <title>RANKING 2.1</title>
    </sec>
    <sec id="sec-4">
      <title>Problem Definition</title>
      <p>In HOCCF, we have a set of n users (U), a set of m items
(I), and two different sets of user feedback, e.g., browses B
and purchases P. Our goal is to find some likely-to-purchase
items from unpurchased items for each user.</p>
      <p>
        In order to fully exploit heterogeneous feedback in
HOCCF such as browses and purchases, we propose not to model
those different feedback jointly as a whole as done in a
recent work [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ], but separately in a staged manner. Specifically,
we model different feedback of a typical user via different
roles such as browser and purchaser. From the perspective
of browser and purchaser, in our role-based Bayesian
personalized ranking (RBPR), we have two tasks of preference
learning, including browser-based preference learning and
purchaser-based preference learning. We illustrate the main
procedure of our proposed solution in Figure 1.
2.2
      </p>
    </sec>
    <sec id="sec-5">
      <title>Browser-based Preference Learning</title>
      <p>In the first step, we assume that a typical user is first
a browser before he/she is converted to a purchaser. And
thus, in our first task, we focus on answering the question
of “whether a user will browse an item”.</p>
      <p>
        In order to address this task, we propose to combine the
two types of one-class feedback, i.e., browses and
purchases, together, and then apply an algorithm for homogeneous
one-class feedback such as BPR [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ], i.e., BPR(B ∪ P).
Mathematically, we will solve the following optimization problem,
min X X X
ΘB∪P u∈U i∈(Bu∪Pu) j∈I\(Bu∪Pu)
fuij ,
(1)
where Bu and Pu are item sets browsed and purchased by
user u, respectively, fuij is the tentative objective function
for a randomly sampled triple (u, i, j), and ΘB∪P denotes
the set of model parameters to be learned [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ].
      </p>
      <p>Once we have learned the model parameters, we can
generate a candidate list of items that a user is likely to browse.
Specifically, for a top-K recommended problem, we will
generate 3K items in this step, so that the refinement in next
step may have more room for improvement.
2.3</p>
    </sec>
    <sec id="sec-6">
      <title>Purchaser-based Preference Learning</title>
      <p>In the second step, we assume that a user will most
likely choose an item from the candidate list that he/she has
browsed. For this reason, in our second task, we mainly
answer the question of “whether a user will purchase an item”.
Input: Users’ browses B and purchases P.</p>
      <p>Output: Top-K recommended items for each user.
Step 1. Conduct browser-based preference learning
via BPR(B ∪ P) as shown in Eq.(1) and obtain 3K
candidate items with highest predicted scores.</p>
      <p>Step 2. Conduct purchaser-based preference
learning via BPR(P) as shown in Eq.(2); predict the
scores on the 3K candidate items and refine the list.</p>
      <p>In order to solve this task, we propose to use the purchase
data only to refine the candidate list from the first step.
The reason is that the purchase feedback is more helpful in
answering whether a certain item will be bought by a user.
Due to the fact that a user’s purchase feedback are few, we
may not get good results if we only apply the second step,
i.e., only use the purchase feedback to find items that will
be bought by a user. This phenomenon is also observed in
our empirical studies.</p>
      <p>
        Similarly, we again adopt BPR [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ] for model training, but
use purchase feedback P only. Mathematically, we learn the
model parameters as follows,
min X X X
ΘP u∈U i∈Pu j∈I\Pu
fuij ,
(2)
where ΘP denotes the model parameters to be learned from
the purchase data only.
      </p>
      <p>With the learned model parameters ΘP , we can predict
the preference of each item i in the candidate list of each user
u, and then re-rank the items in the list. The refined list is
expected to better represent the purchase likelihood of a
certain user, i.e., the recommendation may be more accurate,
which is also verified in our empirical studies. We illustrate
the effect of the difference between those two ranked lists in
Figure 1.</p>
      <p>
        For the optimization problems in the aforementioned two
learning tasks, we can apply stochastic gradient descent to
learn the model parameters [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ]. We put the two preference
learning tasks in one single algorithm in Figure 2 in order
to get a complete picture.
3.1
      </p>
    </sec>
    <sec id="sec-7">
      <title>EXPERIMENTAL RESULTS</title>
    </sec>
    <sec id="sec-8">
      <title>Datasets and Evaluation Metrics</title>
      <p>
        In our empirical studies, in order to directly compare our
RBPR with the very recent algorithm for HOCCF, i.e.,
TJSL [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]. We first use the three public datasets in [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]1, including
MovieLens 100K (ML100K), MovieLens 1M (ML1M) and
Alibaba2015. The detailed description of those three
data can be found in [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]. We also study the performance of
our RBPR on two large datasets, including MovieLens 10M
(ML10M)2 and Netflix.
      </p>
      <p>ML10M is a public data with about 10 million numerical
ratings in {0.5, 1, 1.5, ..., 4.5, 5}, and Netflix is the dataset
used in the famous $100 Million competition with about 0.1
billion scores in {1, 2, 3, 4, 5}. For both ML10M and Netflix,
we first divide the data into five parts with equal
numbers of (u, i, rui) triples, we then take one part and keep the
(u, i) pairs with rui = 5 as purchases for training, take one
part and keep the (u, i) pairs with rui = 5 as purchases for
validation, and take one part and keep the (u, i) pairs with
rui = 5 as purchases for test, and finally take the remaining
two parts and keep all the (u, i) pairs as browses. We repeat
this procedure for three times in order to obtain three copies
of data.</p>
      <p>We put the statistics of the datasets in Table 1.</p>
      <p>For evaluation, we use five ranking-oriented metrics,
including Precision@5, Recall@5, F1@5, NDCG@5 and 1-call@5.
3.2</p>
    </sec>
    <sec id="sec-9">
      <title>Baselines and Parameter Settings</title>
      <p>
        Because HOCCF is a relatively new recommendation
problem, very few solutions have been proposed. In our empirical
studies, we thus include the very recent algorithm TJSL [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]
for HOCCF and also the seminal work BPR [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ] for OCCF.
• BPR (Bayesian personalized ranking) is an
efficient and accurate recommendation algorithm for
homogeneous one-class feedback such as purchases, which
mines users’ preferences by assuming that a user
prefers a purchased item to an unpurchased item.
• TJSL (transfer via joint similarity learning) is the
stateof-the-art method for heterogeneous one-class feedback
such as browses and purchases, which jointly learns the
similarity between a candidate item and a purchased
item, and the similarity between a candidate item and
a likely-to-purchase item.
1http://www.cse.ust.hk/∼weikep/TL4HOCCF/
2http://grouplens.org/datasets/movielens/10m/
For BPR, TJSL and RBPR, we fix the dimension as d =
20 and the learning rate as γ = 0.01. For BPR and TJSL
on ML100K, ML1M and Alibaba2015, we directly use the
results from [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]. For RBPR on all the datasets and BPR
on ML10M and Netflix, we search the best tradeoff
parameter from {0.001, 0.01, 0.1} and iteration number from
{100, 500, 1000} via NDCG@15. In order to make the
results consistent and comparable with [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ], we run five times
of RBPR on ML100K, ML1M and Alibaba2015, and report
the averaged performance. For ML10M and Netflix, we
report the averaged results on three copies of data.
3.3
      </p>
    </sec>
    <sec id="sec-10">
      <title>Results</title>
      <p>We report the recommendation performance in Table 2.
We can have the following observations:
• RBPR and TJSL are better than BPR in all cases
including five evaluation metrics and five datasets, which
clearly shows that the feedback browses are useful for
learning and mining users’ hidden preferences, and RBPR
and TJSL are able to make use of users’ heterogeneous
feedback well.
• RBPR and TJSL are comparable on three small
datasets, e.g., TJSL is the best on ML100K, RBPR is the best
on ML1M, and TJSL and RBPR are comparable on
Alibaba2015.
• TJSL is too slow to generate recommendations on
two large datasets within 24 hours, while RBPR can
produce significantly better results than BPR, which
shows that our RBPR is a more practical solution
regarding the efficiency.</p>
      <p>The overall performance in Table 2 shows that our RBPR
performs the best in making use of the heterogeneous
oneclass feedback.</p>
      <p>In order to check the performance improvement of our
two-stage role-based preference learning solution, we also
check the performance of the generated candidate items as
shown in Figure 1. Specifically, we denote the method for
generating the candidates as RBPR(Browser) since it is based
on the role of browser only, and the final recommendation as
RBPR(Browser,Purchaser). We report the performance on
Precision and NDCG in Figure 3 (other metrics are similar),
from which we can see that the second stage of candidate
refinement using the purchase data can significantly improve
the performance. The improvement also verifies our main
assumption that there are usually two separate stages for a
user’s shopping action, i.e., browse and purchase.</p>
      <p>Method
BPR
TJSL
RBPR
BPR
TJSL
RBPR
BPR
TJSL
RBPR
BPR
TJSL
RBPR
BPR
TJSL
RBPR
0.0552±0.0006
0.0697±0.0016
0.0654±0.0013
0.0928±0.0008
0.1012±0.0011
0.1086±0.0009
0.0050±0.0006
0.0071±0.0004
0.0076±0.0005
RBPR(Browser)
RBPR(Browser,Purchaser)
RBPR(Browser)
RBPR(Browser,Purchaser)
0 ML100K</p>
      <p>ML1M Alibaba2015 ML10M</p>
      <p>Dataset</p>
      <p>Netflix</p>
    </sec>
    <sec id="sec-11">
      <title>4. CONCLUSIONS AND FUTURE WORK</title>
      <p>
        In this paper, we study an important recommendation
problem called heterogeneous one-class collaborative
filtering (HOCCF) from a novel perspective of users’ roles.
Specifically, we propose a novel role-based preference learning
framework, i.e., role-based Bayesian personalized ranking (RBPR),
based on a seminal work [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ]. Extensive empirical studies
show that our RBPR is more accurate than the seminal work
for OCCF, i.e., BPR [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ], and a very recent similarity
learning algorithm for HOCCF, i.e., TJSL [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ]. Furthermore, our
RBPR is very efficient with the inherited merits of BPR [
        <xref ref-type="bibr" rid="ref4">4</xref>
        ],
while TJSL [
        <xref ref-type="bibr" rid="ref3">3</xref>
        ] is difficult to produce recommendations on
two large datasets.
      </p>
      <p>For future work, we are interested in extending and
applying our role-based preference learning framework to
other recommendation settings with more types of users’ roles
such as searcher, browser, purchaser, rater and friends.</p>
    </sec>
  </body>
  <back>
    <ref-list>
      <ref id="ref1">
        <mixed-citation>
          [1]
          <string-name>
            <given-names>S.</given-names>
            <surname>Kabbur</surname>
          </string-name>
          ,
          <string-name>
            <given-names>X.</given-names>
            <surname>Ning</surname>
          </string-name>
          , and
          <string-name>
            <given-names>G.</given-names>
            <surname>Karypis</surname>
          </string-name>
          . Fism:
          <article-title>Factored item similarity models for top-n recommender systems</article-title>
          .
          <source>In Proceedings of the 19th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, KDD '13</source>
          , pages
          <fpage>659</fpage>
          -
          <lpage>667</lpage>
          ,
          <year>2013</year>
          .
        </mixed-citation>
      </ref>
      <ref id="ref2">
        <mixed-citation>
          [2]
          <string-name>
            <given-names>R.</given-names>
            <surname>Pan</surname>
          </string-name>
          ,
          <string-name>
            <given-names>Y.</given-names>
            <surname>Zhou</surname>
          </string-name>
          ,
          <string-name>
            <given-names>B.</given-names>
            <surname>Cao</surname>
          </string-name>
          ,
          <string-name>
            <given-names>N. N.</given-names>
            <surname>Liu</surname>
          </string-name>
          ,
          <string-name>
            <given-names>R.</given-names>
            <surname>Lukose</surname>
          </string-name>
          ,
          <string-name>
            <given-names>M.</given-names>
            <surname>Scholz</surname>
          </string-name>
          , and
          <string-name>
            <given-names>Q.</given-names>
            <surname>Yang</surname>
          </string-name>
          .
          <article-title>One-class collaborative filtering</article-title>
          .
          <source>In Proceedings of the 8th IEEE International Conference on Data Mining, ICDM '08</source>
          , pages
          <fpage>502</fpage>
          -
          <lpage>511</lpage>
          ,
          <year>2008</year>
          .
        </mixed-citation>
      </ref>
      <ref id="ref3">
        <mixed-citation>
          [3]
          <string-name>
            <given-names>W.</given-names>
            <surname>Pan</surname>
          </string-name>
          , M. Liu, and
          <string-name>
            <given-names>Z.</given-names>
            <surname>Ming</surname>
          </string-name>
          .
          <article-title>Transfer learning for heterogeneous one-class collaborative filtering</article-title>
          .
          <source>IEEE Intelligent Systems</source>
          ,
          <year>2016</year>
          .
        </mixed-citation>
      </ref>
      <ref id="ref4">
        <mixed-citation>
          [4]
          <string-name>
            <given-names>S.</given-names>
            <surname>Rendle</surname>
          </string-name>
          ,
          <string-name>
            <given-names>C.</given-names>
            <surname>Freudenthaler</surname>
          </string-name>
          ,
          <string-name>
            <given-names>Z.</given-names>
            <surname>Gantner</surname>
          </string-name>
          , and L.
          <string-name>
            <surname>Schmidt-Thieme</surname>
          </string-name>
          .
          <article-title>Bpr: Bayesian personalized ranking from implicit feedback</article-title>
          .
          <source>In Proceedings of the 25th Conference on Uncertainty in Artificial Intelligence, UAI '09</source>
          , pages
          <fpage>452</fpage>
          -
          <lpage>461</lpage>
          ,
          <year>2009</year>
          .
        </mixed-citation>
      </ref>
    </ref-list>
  </back>
</article>