Using Machine Learning to Predict Links and Improve Steiner Tree Solutions to Team Formation Problems

  • Peter KeaneEmail author
  • Faisal Ghaffar
  • David Malone
Conference paper
Part of the Studies in Computational Intelligence book series (SCI, volume 882)


The team formation problem has existed for many years in various guises. One important problem in the team formation problem is to produce small teams that have a required set of skills. We propose a framework that incorporates machine learning to predict unobserved links between collaborators, alongside improved Steiner tree problems to form small teams to cover given tasks. Our framework not only considers size of the team but also how likely are team members are going to collaborate with each other. The results show that this model consistently returns smaller collaborative teams.


Team formation Link prediction Steiner tree 



This publication has emanated from research conducted with the financial support of Science Foundation Ireland (SFI) and is co-funded under the European Regional Development Fund under Grant Number 13/RC/2077.


  1. 1.
    Hasan, M.A., Chaoji, V., Salem, S., Zaki, M.: Link prediction using supervised learning. In: SDM 2006: Workshop on Link Analysis, Counter-Terrorism and Security (2006)Google Scholar
  2. 2.
    Bollobás, B.: Modern Graph Theory, vol. 184. Springer, New York (2013)zbMATHGoogle Scholar
  3. 3.
    Chawla, N.V., Bowyer, K.W., Hall, L.O., Philip Kegelmeyer, W.: Smote: synthetic minority over-sampling technique. J. Artif. Intell. Res. 16, 321–357 (2002)CrossRefGoogle Scholar
  4. 4.
    Gorla, N., Lam, Y.W.: Who should work with whom?: building effective software project teams. Commun. ACM 47(6), 79–82 (2004)CrossRefGoogle Scholar
  5. 5.
    Hwang, F.K., Richards, D.S., Winter, P.: The Steiner Tree Problem, vol. 53. Elsevier, Amsterdam (1992)zbMATHGoogle Scholar
  6. 6.
    Lappas, T., Liu, K., Terzi, E.: Finding a team of experts in social networks. In: Proceedings of the 15th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 467–476. ACM (2009)Google Scholar
  7. 7.
    Lü, L., Zhou, T.: Link prediction in complex networks: a survey. Phys. A Stat. Mech. Appl. 390(6), 1150–1170 (2011)CrossRefGoogle Scholar
  8. 8.
    Newman, M.E.J.: Clustering and preferential attachment in growing networks. Phys. Rev. E 64(2), 025102 (2001)CrossRefGoogle Scholar
  9. 9.
    Pendharkar, P.C., Rodger, J.A.: The relationship between software development team size and software development cost. Commun. ACM 52(1), 141–144 (2009)CrossRefGoogle Scholar
  10. 10.
    Sharma, R., McAreavey, K., Hong, J., Ghaffar, F.: Individual-level social capital in weighted and attributed social networks. In: 2018 IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining (ASONAM), pp. 1032–1037. IEEE (2018)Google Scholar
  11. 11.
    Spadon, G., de Carvalho, A.C.P.L.F., Rodrigues-Jr, J.F., Alves, L.G.A.: Reconstructing commuters network using machine learning and urban indicators. Sci. Rep. 9(1), 1–13 (2019)Google Scholar
  12. 12.
    Tang, J., Wu, S., Sun, J., Su, H.: Cross-domain collaboration recommendation. In: Proceedings of the 18th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 1285–1293. ACM (2012)Google Scholar
  13. 13.
    Wang, X., Zhao, Z., Ng, W.: A comparative study of team formation in social networks. In: International Conference on Database Systems for Advanced Applications, pp. 389–404. Springer (2015)Google Scholar
  14. 14.
    Watts, D.J., Strogatz, S.H.: Collective dynamics of ‘small-world’ networks. Nature 393(6684), 440 (1998)CrossRefGoogle Scholar
  15. 15.
    Wu, S., Sun, J., Tang, J.: Patent partner recommendation in enterprise social networks. In: Proceedings of the Sixth ACM International Conference on Web Search and Data Mining, pp. 43–52. ACM (2013)Google Scholar
  16. 16.
    Zhang, J., Lv, Y., Yu, P.: Enterprise social link recommendation. In: Proceedings of the 24th ACM International on Conference on Information and Knowledge Management, pp. 841–850. ACM (2015)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2020

Authors and Affiliations

  1. 1.Maynooth UniversityMaynoothIreland
  2. 2.Innovation Exchange, IBM IrelandDublinIreland

Personalised recommendations